Stable-Video-Diffusion-Img2Vid

Running on Zero

Fabrice-TIERCELIN commited on Jun 19

Commit

992a99c

•

1 Parent(s): 89df602

Need to use a public model (#7)

- Need to use a public model (eb5d6e2387bb6f2206ffe449557051d9c4b7b003)

Co-authored-by: Fabrice TIERCELIN <Fabrice-TIERCELIN@users.noreply.huggingface.co>

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ import spaces
 #gradio.helpers.CACHED_FOLDER = '/data/cache'
 pipe = StableVideoDiffusionPipeline.from_pretrained(
-    "multimodalart/stable-video-diffusion", torch_dtype=torch.float16, variant="fp16"
 )
 pipe.to("cuda")
 #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)

 #gradio.helpers.CACHED_FOLDER = '/data/cache'
 pipe = StableVideoDiffusionPipeline.from_pretrained(
+    "vdo/stable-video-diffusion-img2vid-xt-1-1", torch_dtype=torch.float16, variant="fp16"
 )
 pipe.to("cuda")
 #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)