Spaces:

multimodalart
/

self-forcing

Running on Zero

multimodalart HF Staff commited on 2 days ago

Commit

e144d2f

verified ·

1 Parent(s): 80a1151

Update wan/modules/t5.py

Files changed (1) hide show

wan/modules/t5.py CHANGED Viewed

@@ -475,7 +475,7 @@ class T5EncoderModel:
         self,
         text_len,
         dtype=torch.bfloat16,
-        device=torch.cuda.current_device(),
         checkpoint_path=None,
         tokenizer_path=None,
         shard_fn=None,

         self,
         text_len,
         dtype=torch.bfloat16,
+        device="cuda",
         checkpoint_path=None,
         tokenizer_path=None,
         shard_fn=None,