Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from diffusers import DiffusionPipeline
|
|
3 |
import torch
|
4 |
|
5 |
# Cargar el modelo FLUX.1-schnell
|
6 |
-
modelo = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell"
|
7 |
modelo.enable_model_cpu_offload() # Para ahorrar memoria en la GPU si es necesario
|
8 |
|
9 |
# Función para generar la imagen con una semilla específica
|
|
|
3 |
import torch
|
4 |
|
5 |
# Cargar el modelo FLUX.1-schnell
|
6 |
+
modelo = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell")
|
7 |
modelo.enable_model_cpu_offload() # Para ahorrar memoria en la GPU si es necesario
|
8 |
|
9 |
# Función para generar la imagen con una semilla específica
|