Spaces:

JoeS51
/

Animate-Diff-vidgen

Running

JoeS51 commited on 25 days ago

Commit

eb0262e

1 Parent(s): 280b1ae

fixed model issue

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,8 @@ import tempfile
 import os
 pipe = AnimateDiffPipeline.from_pretrained(
-    "guoyww/animatediff", torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
 )
 pipe.to("cuda" if torch.cuda.is_available() else "cpu")

 import os
 pipe = AnimateDiffPipeline.from_pretrained(
+    "ByteDance/AnimateDiff-Lightning",
+    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
 )
 pipe.to("cuda" if torch.cuda.is_available() else "cpu")