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
|