multimodalart HF staff commited on
Commit
41915d4
1 Parent(s): 7adec1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ pipe = StableVideoDiffusionPipeline.from_pretrained(
20
  "multimodalart/stable-video-diffusion", torch_dtype=torch.float16, variant="fp16"
21
  )
22
  pipe.to("cuda")
23
- pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
24
  #pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=True)
25
 
26
  max_64_bit_int = 2**63 - 1
 
20
  "multimodalart/stable-video-diffusion", torch_dtype=torch.float16, variant="fp16"
21
  )
22
  pipe.to("cuda")
23
+ #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
24
  #pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=True)
25
 
26
  max_64_bit_int = 2**63 - 1