Rooni commited on
Commit
c9f5fc9
1 Parent(s): 5bb8bb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from diffusers import DiffusionPipeline
4
  # Загрузка модели
5
  model_id = "ali-vilab/text-to-video-ms-1.7b"
6
  pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16")
7
- pipe = pipe.to("cuda")
8
 
9
  def generate_video(prompt: str, num_inference_steps: int = 50, guidance_scale: float = 7.5):
10
  """
 
4
  # Загрузка модели
5
  model_id = "ali-vilab/text-to-video-ms-1.7b"
6
  pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16")
7
+ pipe = pipe.to("cpu") # Установка CPU как устройства для модели
8
 
9
  def generate_video(prompt: str, num_inference_steps: int = 50, guidance_scale: float = 7.5):
10
  """