Spaces:

rayli
/

Puppet-Master

Running on Zero

liray commited on Aug 17, 2024

Commit

cf28fff

1 Parent(s): a61fc84

Fix.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -198,8 +198,9 @@ sam_predictor = sam_init()
 model = model_init()
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     "stabilityai/stable-video-diffusion-img2vid",
-    torch_dtype=torch.float16, variant="fp16"
-).to("cuda")
 @spaces.GPU(duration=10)

 model = model_init()
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     "stabilityai/stable-video-diffusion-img2vid",
+    torch_dtype=torch.float16, variant="fp16", device="cpu"
+)
+pipe = pipe.to("cuda")
 @spaces.GPU(duration=10)