Spaces:
Running
on
Zero
Running
on
Zero
Fix.
Browse files
app.py
CHANGED
|
@@ -197,11 +197,9 @@ def model_init():
|
|
| 197 |
sam_predictor = sam_init()
|
| 198 |
model = model_init()
|
| 199 |
pipe = StableVideoDiffusionPipeline.from_pretrained(
|
| 200 |
-
"/
|
| 201 |
torch_dtype=torch.float16, variant="fp16"
|
| 202 |
-
)
|
| 203 |
-
pipe.vae.to(dtype=torch.float16, device="cuda")
|
| 204 |
-
pipe.image_encoder = pipe.image_encoder.to("cuda")
|
| 205 |
|
| 206 |
|
| 207 |
@spaces.GPU(duration=10)
|
|
|
|
| 197 |
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 |
+
).to("cuda")
|
|
|
|
|
|
|
| 203 |
|
| 204 |
|
| 205 |
@spaces.GPU(duration=10)
|