Spaces:

pva22
/

course_diffusion

Sleeping

pva22 commited on Jan 18

Commit

ae447d8

1 Parent(s): e9ce74f

cpu

Files changed (2) hide show

.DS_Store ADDED Viewed

Binary file (6.15 kB). View file

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import torch
 # Загрузка модели
 def load_model(model_id):
     pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-    pipe.to("cuda")
     return pipe
 # Генерация изображения

 # Загрузка модели
 def load_model(model_id):
     pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+    pipe.to("cpu")
     return pipe
 # Генерация изображения