ai-tube-model-als-10

Running on A10G

wangfuyun commited on Feb 27

Commit

0e86fc7

•

1 Parent(s): d2a129d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -55,7 +55,7 @@ pipe = StableVideoDiffusionPipeline.from_pretrained(
     variant="fp16",
 )
 pipe.to("cuda")
-# pipe.enable_model_cpu_offload()  # for smaller cost
 model_select("AnimateLCM-SVD-xt-1.1.safetensors")
 # pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True) # for faster inference

     variant="fp16",
 )
 pipe.to("cuda")
+pipe.enable_model_cpu_offload()  # for smaller cost
 model_select("AnimateLCM-SVD-xt-1.1.safetensors")
 # pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True) # for faster inference