jbilcke-hf HF staff commited on
Commit
fde41ed
1 Parent(s): 1298d15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -181,9 +181,9 @@ pipe = StableVideoDiffusionPipeline.from_pretrained(
181
  variant="fp16",
182
  )
183
  pipe.to("cuda")
184
- pipe.enable_model_cpu_offload() # for smaller cost
185
  model_select("AnimateLCM-SVD-xt-1.1.safetensors")
186
- # pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True) # for faster inference
187
 
188
 
189
  max_64_bit_int = 2**63 - 1
 
181
  variant="fp16",
182
  )
183
  pipe.to("cuda")
184
+ # pipe.enable_model_cpu_offload() # for smaller cost
185
  model_select("AnimateLCM-SVD-xt-1.1.safetensors")
186
+ pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True) # for faster inference
187
 
188
 
189
  max_64_bit_int = 2**63 - 1