Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,10 +4,10 @@ import torch
|
|
4 |
|
5 |
## Cargar el modelo
|
6 |
model_id = "stablediffusionapi/epicrealism-natural-sin-r" # Reemplazar con el ID de tu modelo en Hugging Face
|
7 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
|
8 |
-
|
9 |
-
pipe = pipe.to("cuda") # Asegúrate de tener una GPU disponible o cambia esto a "cpu"
|
10 |
-
|
11 |
|
12 |
def inferencia(prompt):
|
13 |
# Realizar la inferencia
|
|
|
4 |
|
5 |
## Cargar el modelo
|
6 |
model_id = "stablediffusionapi/epicrealism-natural-sin-r" # Reemplazar con el ID de tu modelo en Hugging Face
|
7 |
+
#pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
|
8 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
|
9 |
+
#pipe = pipe.to("cuda") # Asegúrate de tener una GPU disponible o cambia esto a "cpu"
|
10 |
+
pipe = pipe.to("cpu") # Asegúrate de tener una GPU disponible o cambia esto a "cpu"
|
11 |
|
12 |
def inferencia(prompt):
|
13 |
# Realizar la inferencia
|