diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-01 10:35:38 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-01 10:35:38 +0300 |
| commit | 11d432d92d63660c516540dcb48faac87669b4f0 (patch) | |
| tree | 2eea3379cd521ac0838bd97c38daf36bea0ed079 /modules | |
| parent | 16b9661d2741b241c3964fcbd56559c078b84822 (diff) | |
add refresh buttons to checkpoint merger
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/ui.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/modules/ui.py b/modules/ui.py index c7b8ea5d..4cc2ce4f 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -1167,8 +1167,14 @@ def create_ui(): with gr.Row():
primary_model_name = gr.Dropdown(modules.sd_models.checkpoint_tiles(), elem_id="modelmerger_primary_model_name", label="Primary model (A)")
+ create_refresh_button(primary_model_name, modules.sd_models.list_models, lambda: {"choices": modules.sd_models.checkpoint_tiles()}, "refresh_checkpoint_A")
+
secondary_model_name = gr.Dropdown(modules.sd_models.checkpoint_tiles(), elem_id="modelmerger_secondary_model_name", label="Secondary model (B)")
+ create_refresh_button(secondary_model_name, modules.sd_models.list_models, lambda: {"choices": modules.sd_models.checkpoint_tiles()}, "refresh_checkpoint_B")
+
tertiary_model_name = gr.Dropdown(modules.sd_models.checkpoint_tiles(), elem_id="modelmerger_tertiary_model_name", label="Tertiary model (C)")
+ create_refresh_button(tertiary_model_name, modules.sd_models.list_models, lambda: {"choices": modules.sd_models.checkpoint_tiles()}, "refresh_checkpoint_C")
+
custom_name = gr.Textbox(label="Custom Name (Optional)")
interp_amount = gr.Slider(minimum=0.0, maximum=1.0, step=0.05, label='Multiplier (M) - set to 0 to get model A', value=0.3)
interp_method = gr.Radio(choices=["Weighted sum", "Add difference"], value="Weighted sum", label="Interpolation Method")
|