profaker commited on
Commit
ef85702
·
verified ·
1 Parent(s): 3581316

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -22,7 +22,6 @@ def predict(prompt, negative_prompt, guidance_scale, num_inference_steps,model,
22
  pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V5.1_noVAE", vae=vae).to("cuda")
23
  if model == "Realistic_V5.0":
24
  pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE", vae=vae).to("cuda")
25
- pipeline.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
26
  if model == "EpicRealism":
27
  pipeline = diffusers.DiffusionPipeline.from_pretrained("emilianJR/epiCRealism", vae=vae).to("cuda")
28
  pipeline.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
 
22
  pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V5.1_noVAE", vae=vae).to("cuda")
23
  if model == "Realistic_V5.0":
24
  pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE", vae=vae).to("cuda")
 
25
  if model == "EpicRealism":
26
  pipeline = diffusers.DiffusionPipeline.from_pretrained("emilianJR/epiCRealism", vae=vae).to("cuda")
27
  pipeline.safety_checker = lambda images, **kwargs: (images, [False] * len(images))