profaker commited on
Commit
652265d
·
verified ·
1 Parent(s): 6f7ff2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -29,6 +29,7 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, s
29
  pipe = AutoPipelineForInpainting.from_pretrained("emilianJR/epiCRealism", vae=vae).to("cuda")
30
  if model == "Realistic_V6.0":
31
  pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE", vae=vae).to("cuda")
 
32
 
33
 
34
  if negative_prompt == "":
 
29
  pipe = AutoPipelineForInpainting.from_pretrained("emilianJR/epiCRealism", vae=vae).to("cuda")
30
  if model == "Realistic_V6.0":
31
  pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE", vae=vae).to("cuda")
32
+ pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
33
 
34
 
35
  if negative_prompt == "":