Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ pipe = StableVideoDiffusionPipeline.from_pretrained(
|
|
55 |
variant="fp16",
|
56 |
)
|
57 |
pipe.to("cuda")
|
58 |
-
|
59 |
model_select("AnimateLCM-SVD-xt-1.1.safetensors")
|
60 |
# pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True) # for faster inference
|
61 |
|
|
|
55 |
variant="fp16",
|
56 |
)
|
57 |
pipe.to("cuda")
|
58 |
+
pipe.enable_model_cpu_offload() # for smaller cost
|
59 |
model_select("AnimateLCM-SVD-xt-1.1.safetensors")
|
60 |
# pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True) # for faster inference
|
61 |
|