Manjushri commited on
Commit
3c9e972
·
verified ·
1 Parent(s): 40d434e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ torch.cuda.empty_cache()
18
  def genie(src_image):
19
  torch.cuda.max_memory_allocated(device=device)
20
  torch.cuda.empty_cache()
21
- pipe = StableVideoDiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid-xt", variant="fp16", use_safetensors=True)
22
  #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
23
 
24
  pipe.enable_xformers_memory_efficient_attention()
 
18
  def genie(src_image):
19
  torch.cuda.max_memory_allocated(device=device)
20
  torch.cuda.empty_cache()
21
+ pipe = StableVideoDiffusionPipeline.from_pretrained("multimodalart/stable-video-diffusion", variant="fp16", use_safetensors=True)
22
  #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
23
 
24
  pipe.enable_xformers_memory_efficient_attention()