ML-INTA commited on
Commit
b048535
1 Parent(s): 9e67631

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("cuda")
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