Update app.py
Browse files
app.py
CHANGED
|
@@ -3,7 +3,9 @@ import torch
|
|
| 3 |
from diffusers import I2VGenXLPipeline
|
| 4 |
from diffusers.utils import export_to_gif, load_image
|
| 5 |
import tempfile
|
|
|
|
| 6 |
|
|
|
|
| 7 |
def initialize_pipeline():
|
| 8 |
# Initialize the pipeline without CUDA support
|
| 9 |
pipeline = I2VGenXLPipeline.from_pretrained("ali-vilab/i2vgen-xl", torch_dtype=torch.float16, variant="fp16")
|
|
|
|
| 3 |
from diffusers import I2VGenXLPipeline
|
| 4 |
from diffusers.utils import export_to_gif, load_image
|
| 5 |
import tempfile
|
| 6 |
+
import spaces
|
| 7 |
|
| 8 |
+
@spaces.GPU
|
| 9 |
def initialize_pipeline():
|
| 10 |
# Initialize the pipeline without CUDA support
|
| 11 |
pipeline = I2VGenXLPipeline.from_pretrained("ali-vilab/i2vgen-xl", torch_dtype=torch.float16, variant="fp16")
|