brightlembo commited on
Commit
72093e1
·
verified ·
1 Parent(s): e0b7f86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,9 +3,9 @@ import gradio as gr
3
  from diffusers import DiffusionPipeline
4
  from diffusers.utils import export_to_video
5
 
6
- # Charger le modèle de diffusion
7
  pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
8
- pipe = pipe.to("cuda")
9
 
10
  # Fonction pour générer la vidéo
11
  def generate_video(prompt):
 
3
  from diffusers import DiffusionPipeline
4
  from diffusers.utils import export_to_video
5
 
6
+ # Charger le modèle de diffusion sans utiliser CUDA
7
  pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
8
+ pipe = pipe.to("cpu") # Utiliser le CPU au lieu du GPU
9
 
10
  # Fonction pour générer la vidéo
11
  def generate_video(prompt):