Spaces:
Running
on
Zero
Running
on
Zero
Fix.
Browse files
app.py
CHANGED
@@ -198,8 +198,9 @@ sam_predictor = sam_init()
|
|
198 |
model = model_init()
|
199 |
pipe = StableVideoDiffusionPipeline.from_pretrained(
|
200 |
"stabilityai/stable-video-diffusion-img2vid",
|
201 |
-
torch_dtype=torch.float16, variant="fp16"
|
202 |
-
)
|
|
|
203 |
|
204 |
|
205 |
@spaces.GPU(duration=10)
|
|
|
198 |
model = model_init()
|
199 |
pipe = StableVideoDiffusionPipeline.from_pretrained(
|
200 |
"stabilityai/stable-video-diffusion-img2vid",
|
201 |
+
torch_dtype=torch.float16, variant="fp16", device="cpu"
|
202 |
+
)
|
203 |
+
pipe = pipe.to("cuda")
|
204 |
|
205 |
|
206 |
@spaces.GPU(duration=10)
|