Update app.py
Browse files
app.py
CHANGED
|
@@ -69,10 +69,10 @@ def ia_imagenes(modelo, prompt, prompt_negativo, uploaded_file, my_strength, my_
|
|
| 69 |
|
| 70 |
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
|
| 71 |
model_id,
|
| 72 |
-
revision="fp16" if torch.cuda.is_available() else "fp32",
|
| 73 |
torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
|
| 74 |
requires_safety_checker = False
|
| 75 |
-
).to("
|
| 76 |
|
| 77 |
my_strength = 0.8 if my_strength == 0 or my_strength == None else my_strength
|
| 78 |
my_guidance_scale = 7.5 if my_guidance_scale == 0 or my_guidance_scale == None else my_guidance_scale
|
|
|
|
| 69 |
|
| 70 |
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
|
| 71 |
model_id,
|
| 72 |
+
#revision="fp16" if torch.cuda.is_available() else "fp32",
|
| 73 |
torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
|
| 74 |
requires_safety_checker = False
|
| 75 |
+
).to("cpu")
|
| 76 |
|
| 77 |
my_strength = 0.8 if my_strength == 0 or my_strength == None else my_strength
|
| 78 |
my_guidance_scale = 7.5 if my_guidance_scale == 0 or my_guidance_scale == None else my_guidance_scale
|