Commit ·
6346496
1
Parent(s): a02443f
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ model_list = ["stabilityai/stable-diffusion-xl-base-1.0",
|
|
| 13 |
model_url_list = ["stabilityai/stable-diffusion-xl-base-1.0/blob/main/sd_xl_base_1.0.safetensors",
|
| 14 |
"Krebzonide/Colossus_Project_XL/blob/main/colossusProjectXLSFW_v202BakedVAE.safetensors",
|
| 15 |
"Krebzonide/Sevenof9_v3_sdxl/blob/main/nsfwSevenof9V3_nsfwSevenof9V3.safetensors"]
|
| 16 |
-
current_model =
|
| 17 |
|
| 18 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
| 19 |
pipe = None
|
|
@@ -94,6 +94,6 @@ with gr.Blocks(css=css) as demo:
|
|
| 94 |
submit_btn.click(generate, [prompt, negative_prompt, samp_steps, guide_scale, batch_size, seed, height, width], [gallery], queue=True)
|
| 95 |
change_model_btn.click(set_base_model, [model_id], queue = False)
|
| 96 |
|
| 97 |
-
pipe = set_base_model(model_list[
|
| 98 |
demo.queue(1)
|
| 99 |
demo.launch(debug=True)
|
|
|
|
| 13 |
model_url_list = ["stabilityai/stable-diffusion-xl-base-1.0/blob/main/sd_xl_base_1.0.safetensors",
|
| 14 |
"Krebzonide/Colossus_Project_XL/blob/main/colossusProjectXLSFW_v202BakedVAE.safetensors",
|
| 15 |
"Krebzonide/Sevenof9_v3_sdxl/blob/main/nsfwSevenof9V3_nsfwSevenof9V3.safetensors"]
|
| 16 |
+
current_model = -1
|
| 17 |
|
| 18 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
| 19 |
pipe = None
|
|
|
|
| 94 |
submit_btn.click(generate, [prompt, negative_prompt, samp_steps, guide_scale, batch_size, seed, height, width], [gallery], queue=True)
|
| 95 |
change_model_btn.click(set_base_model, [model_id], queue = False)
|
| 96 |
|
| 97 |
+
pipe = set_base_model(model_list[0])
|
| 98 |
demo.queue(1)
|
| 99 |
demo.launch(debug=True)
|