Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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("
|
| 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 |
# Функция для генерации изображения
|