Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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))
|