wangfuyun commited on
Commit
cb9268e
1 Parent(s): 26476aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -54,9 +54,9 @@ pipe = StableVideoDiffusionPipeline.from_pretrained(
54
  variant="fp16",
55
  )
56
  pipe.to("cuda")
57
- # pipe.enable_model_cpu_offload() # for smaller cost
58
  model_select("AnimateLCM-SVD-xt.safetensors")
59
- pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True) # for faster inference
60
 
61
 
62
  max_64_bit_int = 2**63 - 1
 
54
  variant="fp16",
55
  )
56
  pipe.to("cuda")
57
+ pipe.enable_model_cpu_offload() # for smaller cost
58
  model_select("AnimateLCM-SVD-xt.safetensors")
59
+ # pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True) # for faster inference
60
 
61
 
62
  max_64_bit_int = 2**63 - 1