Rooni commited on
Commit
261c072
·
1 Parent(s): 147f8ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ model_paths = {
12
  models = {}
13
  for name, path in model_paths.items():
14
  pipeline = StableDiffusionPipeline.from_pretrained(path)
15
- pipeline.to("cuda")
16
  models[name] = pipeline
17
 
18
  # Функция для генерации изображения
 
12
  models = {}
13
  for name, path in model_paths.items():
14
  pipeline = StableDiffusionPipeline.from_pretrained(path)
15
+ pipeline.to("cpu")
16
  models[name] = pipeline
17
 
18
  # Функция для генерации изображения