comfyanonymous commited on
Commit
2f9474a
1 Parent(s): e050150

Update convert.py

Browse files
Files changed (1) hide show
  1. 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").cuda()
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()))