Spaces:
Build error
Build error
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")
|