Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4993 from vladmandic/feat/secondary-unet-networks

Browse Source 
feat(ui): load unet into secondary slot from networks panel
This commit is contained in:
Vladimir Mandic
2026-07-12 09:02:12 +02:00
committed by GitHub
parent cf031b869c 00d1b7cff6
commit 99a963271e
3 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/ui_extra_networks.py
+1 -1
View File
@@ -774,7 +774,7 @@ def create_ui(container, button_parent: gr.Button, tabname: str, skip_indexing =
def ui_tab_change(page):
scan_visible = page in ['Model', 'Lora', 'VAE', 'UNet/DiT', 'Hypernetwork', 'Embedding']
save_visible = page in ['Style']
model_visible = page in ['Model']
model_visible = page in ['Model', 'UNet/DiT']
return [gr.update(visible=scan_visible), gr.update(visible=save_visible), gr.update(visible=model_visible)]
ui.button_refresh = ui_components.ToolButton(ui_symbols.refresh, elem_id=f"{tabname}_extra_refresh")
modules/ui_settings.py
+7
View File
@@ -461,6 +461,13 @@ def create_quicksettings(interfaces):
inputs=[shared.settings_components["sd_unet"], dummy_component],
outputs=[shared.settings_components["sd_unet"], text_settings],
)
button_set_unet_secondary = gr.Button("Change UNet secondary", elem_id="change_unet_secondary", visible=False)
button_set_unet_secondary.click(
fn=lambda value, _: run_settings_single(value, key="sd_unet_secondary"),
_js="consumeDesiredUNetName",
inputs=[shared.settings_components["sd_unet_secondary"], dummy_component],
outputs=[shared.settings_components["sd_unet_secondary"], text_settings],
)
def reference_submit(model):
if '@' not in model: # diffusers
ui/ui.ts
+6 -2
View File
@@ -681,8 +681,12 @@ function getDesiredCheckpointName() {
export function selectUNet(name) {
desiredUNetName = name;
gradioApp().getElementById('change_unet').click();
log(`selectUNet: ${desiredUNetName}`);
const tabName = getENActiveTab();
const btnModel = gradioApp().getElementById(`${tabName}_extra_model`);
const isSecondary = btnModel && btnModel.classList.contains('toolbutton-selected');
if (isSecondary) gradioApp().getElementById('change_unet_secondary').click();
else gradioApp().getElementById('change_unet').click();
log(`selectUNet ${isSecondary ? 'secondary' : 'primary'}: ${desiredUNetName}`);
markSelectedCards([desiredUNetName], 'unet');
}