Update convert.py
Browse files- convert.py +1 -1
convert.py
CHANGED
|
@@ -17,7 +17,7 @@ output = "freeway_animation_demo_hunyuan_dit.safetensors"
|
|
| 17 |
|
| 18 |
|
| 19 |
|
| 20 |
-
mt5 = T5EncoderModel.from_pretrained("google/mt5-xl")
|
| 21 |
tokenizer = T5Tokenizer.from_pretrained("google/mt5-xl")
|
| 22 |
|
| 23 |
sp_model = torch.ByteTensor(list(tokenizer.sp_model.serialized_model_proto()))
|
|
|
|
| 17 |
|
| 18 |
|
| 19 |
|
| 20 |
+
mt5 = T5EncoderModel.from_pretrained("google/mt5-xl")
|
| 21 |
tokenizer = T5Tokenizer.from_pretrained("google/mt5-xl")
|
| 22 |
|
| 23 |
sp_model = torch.ByteTensor(list(tokenizer.sp_model.serialized_model_proto()))
|