HunYuanDiT fix double text encoder input

#58
This commit is contained in:
City
2024-05-31 21:56:37 +02:00
parent 58ecf66b7a
commit 0f83907911
+1 -1
View File
@@ -100,7 +100,7 @@ class HYDiTTextEncode:
# T5 # T5
T5.load_model() T5.load_model()
t5_pre = T5.tokenizer( t5_pre = T5.tokenizer(
text, text_t5,
max_length = T5.cond_stage_model.max_length, max_length = T5.cond_stage_model.max_length,
padding = 'max_length', padding = 'max_length',
truncation = True, truncation = True,