jbilcke-hf HF staff commited on
Commit
48f2aa4
·
verified ·
1 Parent(s): 23224a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -20,7 +20,9 @@ from huggingface_hub import hf_hub_download
20
  SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
21
 
22
  pipe = StableVideoDiffusionPipeline.from_pretrained(
23
- "stabilityai/stable-video-diffusion-img2vid-xt", torch_dtype=torch.float16, variant="fp16"
 
 
24
  )
25
  pipe.to("cuda")
26
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 
20
  SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
21
 
22
  pipe = StableVideoDiffusionPipeline.from_pretrained(
23
+ "stabilityai/stable-video-diffusion-img2vid-xt-1-1",
24
+ torch_dtype=torch.float16,
25
+ variant="fp16"
26
  )
27
  pipe.to("cuda")
28
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)