Krebzonide
commited on
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)
|