CharlieAmalet commited on
Commit
1c5196e
·
verified ·
1 Parent(s): 18ed7af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -24,7 +24,8 @@ import spaces
24
  # login(token=HF_API_KEY)
25
 
26
  pipe = StableVideoDiffusionPipeline.from_pretrained(
27
- "stabilityai/stable-video-diffusion-img2vid-xt-1-1",
 
28
  torch_dtype=torch.float16,
29
  variant="fp16"
30
  )
@@ -32,7 +33,8 @@ pipe.to("cuda")
32
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
33
  #pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=True)
34
 
35
- max_64_bit_int = 2**63 - 1
 
36
 
37
  @spaces.GPU(enable_queue=True)
38
  def generate_video(
 
24
  # login(token=HF_API_KEY)
25
 
26
  pipe = StableVideoDiffusionPipeline.from_pretrained(
27
+ # "stabilityai/stable-video-diffusion-img2vid-xt-1-1",
28
+ "vdo/stable-video-diffusion-img2vid-xt-1-1",
29
  torch_dtype=torch.float16,
30
  variant="fp16"
31
  )
 
33
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
34
  #pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=True)
35
 
36
+ max_64_bit_int =
37
+ 2**63 - 1
38
 
39
  @spaces.GPU(enable_queue=True)
40
  def generate_video(