lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-03 22:58:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update forge_controlnet.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-27 10:15:56 -08:00
parent cac0fdf15c
commit 374ff2ca16
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
extensions-builtin/sd_forge_controlnet/scripts/forge_controlnet.py
View File

@@ -14,19 +14,21 @@ class ControlNetForge(scripts.Script):
return scripts.AlwaysVisible
def build_unit_ui(self):
return
return []
def ui(self, *args, **kwargs):
all_ui_components = []
with gr.Accordion(open=False, label=self.title()):
unit_count = gr.Slider(label='ControlNet Units', minimum=0, maximum=MAX_UNITS, step=1, value=DEFAULT_UNITS)
unit_accordions = []
for i in range(16):
with gr.Accordion(open=True, label=f'ControlNet Unit {i+1}',
visible=i < DEFAULT_UNITS) as unit_accordion:
self.build_unit_ui()
all_ui_components += self.build_unit_ui()
unit_accordions.append(unit_accordion)
unit_count.change(lambda c: [gr.update(visible=i < c) for i in range(MAX_UNITS)],
inputs=unit_count, outputs=unit_accordions, show_progress=False, queue=False)
return [unit_count]
return all_ui_components