aboutsummaryrefslogtreecommitdiff
path: root/modules/ui.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-01-28 22:52:27 +0300
committerAUTOMATIC <16777216c@gmail.com>2023-01-28 22:52:27 +0300
commit1d8e06d542176beade37d2d36cb57460c3c6772f (patch)
treefe43b215e24204cb80f94469af11f6b9e180e4d3 /modules/ui.py
parent91c8d0dcfc9a5d46aec47fd3eb34c633c104b5bb (diff)
add checkpoints tab for extra networks UI
Diffstat (limited to 'modules/ui.py')
-rw-r--r--modules/ui.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/modules/ui.py b/modules/ui.py
index 4e082408..f1195692 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1560,6 +1560,14 @@ def create_ui():
outputs=[component, text_settings],
)
+ button_set_checkpoint = gr.Button('Change checkpoint', elem_id='change_checkpoint', visible=False)
+ button_set_checkpoint.click(
+ fn=lambda value, _: run_settings_single(value, key='sd_model_checkpoint'),
+ _js="function(v){ var res = desiredCheckpointName; desiredCheckpointName = ''; return [res || v, null]; }",
+ inputs=[component_dict['sd_model_checkpoint'], dummy_component],
+ outputs=[component_dict['sd_model_checkpoint'], text_settings],
+ )
+
component_keys = [k for k in opts.data_labels.keys() if k in component_dict]
def get_settings_values():