Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -17,7 +17,7 @@ def read_content(file_path: str) -> str:
|
|
| 17 |
|
| 18 |
def predict(prompt, negative_prompt, guidance_scale, num_inference_steps,model, scheduler, lora, lora_weight):
|
| 19 |
pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/RealVisXL_V4.0", vae=vae).to("cuda")
|
| 20 |
-
if model
|
| 21 |
pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE", vae=vae).to("cuda")
|
| 22 |
pipeline.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
|
| 23 |
if model == "Realistic_V5.1":
|
|
|
|
| 17 |
|
| 18 |
def predict(prompt, negative_prompt, guidance_scale, num_inference_steps,model, scheduler, lora, lora_weight):
|
| 19 |
pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/RealVisXL_V4.0", vae=vae).to("cuda")
|
| 20 |
+
if model == 'Realistic_V6.0':
|
| 21 |
pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE", vae=vae).to("cuda")
|
| 22 |
pipeline.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
|
| 23 |
if model == "Realistic_V5.1":
|