Krebzonide commited on
Commit
6346496
·
1 Parent(s): a02443f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = 0
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[current_model])
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)