+1
-1
@@ -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,
|
||||||
|
|||||||
Reference in New Issue
Block a user