Spaces:

rayli
/

Puppet-Master

Running on Zero

liray commited on Aug 17, 2024

Commit

8fd1be3

1 Parent(s): cf28fff

Fix.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -198,7 +198,7 @@ sam_predictor = sam_init()
 model = model_init()
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     "stabilityai/stable-video-diffusion-img2vid",
-    torch_dtype=torch.float16, variant="fp16", device="cpu"
 )
 pipe = pipe.to("cuda")

 model = model_init()
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     "stabilityai/stable-video-diffusion-img2vid",
+    torch_dtype=torch.float16, variant="fp16", map_location="cpu"
 )
 pipe = pipe.to("cuda")