mrcuddle commited on
Commit
9821031
1 Parent(s): 572c4e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import spaces
8
  pipeline = I2VGenXLPipeline.from_pretrained("ali-vilab/i2vgen-xl", torch_dtype=torch.float16, variant="fp16")
9
  pipeline.enable_model_cpu_offload()
10
 
11
- @spaces.GPU
12
  def generate_gif(image, prompt, negative_prompt, num_inference_steps, guidance_scale, seed):
13
  # Load the image
14
  image = load_image(image).convert("RGB")
 
8
  pipeline = I2VGenXLPipeline.from_pretrained("ali-vilab/i2vgen-xl", torch_dtype=torch.float16, variant="fp16")
9
  pipeline.enable_model_cpu_offload()
10
 
11
+ @spaces.GPU(duration=240)
12
  def generate_gif(image, prompt, negative_prompt, num_inference_steps, guidance_scale, seed):
13
  # Load the image
14
  image = load_image(image).convert("RGB")