From 9f95e6d4812ca7acb3ec56cbf45fbc85aef0236d Mon Sep 17 00:00:00 2001 From: AlUlkesh <99896447+AlUlkesh@users.noreply.github.com> Date: Mon, 11 Sep 2023 14:55:51 +0200 Subject: [PATCH] Fix for controlnet option name change, #234 --- scripts/image_browser.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/scripts/image_browser.py b/scripts/image_browser.py index 1764850..3040462 100644 --- a/scripts/image_browser.py +++ b/scripts/image_browser.py @@ -1254,7 +1254,10 @@ def create_tab(tab: ImageBrowserTab, current_gr_tab: gr.Tab): sendto_controlnet_txt2img = gr.Button("Send to txt2img ControlNet", visible=controlnet) sendto_controlnet_img2img = gr.Button("Send to img2img ControlNet", visible=controlnet) with gr.Row(visible=controlnet): - controlnet_max = opts.data.get("control_net_max_models_num", 1) + if "control_net_unit_count" in opts.data: + controlnet_max = opts.data.get("control_net_unit_count", 3) + else: + controlnet_max = opts.data.get("control_net_max_models_num", 3) sendto_controlnet_num = gr.Dropdown([str(i) for i in range(controlnet_max)], label="ControlNet number", value="0", interactive=True, visible=(controlnet and controlnet_max > 1)) if controlnet_max is None: sendto_controlnet_type = gr.Textbox(value="none", visible=False)