Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,9 @@ from huggingface_hub import hf_hub_download
|
|
20 |
SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
|
21 |
|
22 |
pipe = StableVideoDiffusionPipeline.from_pretrained(
|
23 |
-
"stabilityai/stable-video-diffusion-img2vid-xt",
|
|
|
|
|
24 |
)
|
25 |
pipe.to("cuda")
|
26 |
pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
|
|
|
20 |
SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
|
21 |
|
22 |
pipe = StableVideoDiffusionPipeline.from_pretrained(
|
23 |
+
"stabilityai/stable-video-diffusion-img2vid-xt-1-1",
|
24 |
+
torch_dtype=torch.float16,
|
25 |
+
variant="fp16"
|
26 |
)
|
27 |
pipe.to("cuda")
|
28 |
pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
|