mrcuddle commited on
Commit
0d22f81
·
verified ·
1 Parent(s): 74d09c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
15
  pipe.enable_model_cpu_offload()
16
  pipe.unet.enable_forward_chunking()
17
 
18
- @spaces.GPU
19
  def generate_video(image, seed=42, fps=7, motion_bucket_id=180, noise_aug_strength=0.1):
20
  # Resize the image
21
  image = image.resize((1024, 576))
 
15
  pipe.enable_model_cpu_offload()
16
  pipe.unet.enable_forward_chunking()
17
 
18
+ @spaces.GPU(duration=300)
19
  def generate_video(image, seed=42, fps=7, motion_bucket_id=180, noise_aug_strength=0.1):
20
  # Resize the image
21
  image = image.resize((1024, 576))