Hugging Face's logo

Spaces:
mrcuddle
/
I2VGenXL-Text-Image-To-Video
Build error

App Files Community
mrcuddle commited on
Commit
bf0c187
·
verified ·
1 Parent(s): c7d5c0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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(device):
8
  # Initialize the pipeline with 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(device):
10
  # Initialize the pipeline with CUDA support
11
  pipeline = I2VGenXLPipeline.from_pretrained("ali-vilab/i2vgen-xl", torch_dtype=torch.float16, variant="fp16")