Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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")
|