diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-22 07:17:12 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-22 07:17:12 +0000 |
| commit | 112416d04171e4bee673f0adc9bd3aeba87ec71a (patch) | |
| tree | 52c2f64e2c78895b63c45300bb6b71a5759e22d1 /modules/ui.py | |
| parent | 0792fae078ba362a5119f56d84e3f490a88690ae (diff) | |
| download | stable-diffusion-webui-gfx803-112416d04171e4bee673f0adc9bd3aeba87ec71a.tar.gz stable-diffusion-webui-gfx803-112416d04171e4bee673f0adc9bd3aeba87ec71a.tar.bz2 stable-diffusion-webui-gfx803-112416d04171e4bee673f0adc9bd3aeba87ec71a.zip | |
add option to discard weights in checkpoint merger UI
Diffstat (limited to 'modules/ui.py')
| -rw-r--r-- | modules/ui.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/modules/ui.py b/modules/ui.py index af6dfb21..eb4b7e6b 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -1248,6 +1248,9 @@ def create_ui(): bake_in_vae = gr.Dropdown(choices=["None"] + list(sd_vae.vae_dict), value="None", label="Bake in VAE", elem_id="modelmerger_bake_in_vae")
create_refresh_button(bake_in_vae, sd_vae.refresh_vae_list, lambda: {"choices": ["None"] + list(sd_vae.vae_dict)}, "modelmerger_refresh_bake_in_vae")
+ with FormRow():
+ discard_weights = gr.Textbox(value="", label="Discard weights with matching name", elem_id="modelmerger_discard_weights")
+
with gr.Row():
modelmerger_merge = gr.Button(elem_id="modelmerger_merge", value="Merge", variant='primary')
@@ -1838,6 +1841,7 @@ def create_ui(): checkpoint_format,
config_source,
bake_in_vae,
+ discard_weights,
],
outputs=[
primary_model_name,
|