add option to select hypernetwork modules when creating

2025-08-05 03:32:37 +00:00 · 2022-10-11 18:04:47 +03:00
parent 5ba23cb41f
commit d682444ecc
3 changed files with 6 additions and 4 deletions
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1037,6 +1037,7 @@ def create_ui(wrap_gradio_gpu_call):
                    gr.HTML(value="<p style='margin-bottom: 0.7em'>Create a new hypernetwork</p>")

                    new_hypernetwork_name = gr.Textbox(label="Name")
+                    new_hypernetwork_sizes = gr.CheckboxGroup(label="Modules", value=["768", "320", "640", "1280"], choices=["768", "320", "640", "1280"])

                    with gr.Row():
                        with gr.Column(scale=3):
@@ -1114,6 +1115,7 @@ def create_ui(wrap_gradio_gpu_call):
            fn=modules.hypernetworks.ui.create_hypernetwork,
            inputs=[
                new_hypernetwork_name,
+                new_hypernetwork_sizes,
            ],
            outputs=[
                train_hypernetwork_name,