ML-INTA commited on
Commit
72fc206
·
1 Parent(s): aa0535e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,8 +14,8 @@ import gradio as gr
14
  # Variables globales
15
  # ######################
16
 
17
- model_id = "helenai/runwayml-stable-diffusion-v1-5-ov-fp32"
18
- #model_id = "runwayml/stable-diffusion-v1-5" -> original
19
 
20
  # ######################
21
  # Funciones auxiliares
@@ -43,7 +43,7 @@ def ia_imagenes(modelo, prompt, prompt_negativo, uploaded_file, my_strength, my_
43
 
44
  pipe = StableDiffusionPipeline.from_pretrained(
45
  model_id,
46
- revision="fp16" if torch.cuda.is_available() else "fp32",
47
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
48
  requires_safety_checker = False
49
  ).to("cpu")
 
14
  # Variables globales
15
  # ######################
16
 
17
+ #model_id = "helenai/runwayml-stable-diffusion-v1-5-ov-fp32"
18
+ model_id = "runwayml/stable-diffusion-v1-5" -> original
19
 
20
  # ######################
21
  # Funciones auxiliares
 
43
 
44
  pipe = StableDiffusionPipeline.from_pretrained(
45
  model_id,
46
+ #revision="fp16" if torch.cuda.is_available() else "fp32",
47
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
48
  requires_safety_checker = False
49
  ).to("cpu")