aboutsummaryrefslogtreecommitdiffstats
path: root/modules/ui.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-10-11 15:04:47 +0000
committerAUTOMATIC <16777216c@gmail.com>2022-10-11 15:04:47 +0000
commitd682444ecc99319fbd2b142a12727501e2884ba7 (patch)
tree0b4cb662da210c35c513e95b2786d2eb4082d6df /modules/ui.py
parent5ba23cb41f28f5856a7f64cb0d95e1e94dce90af (diff)
downloadstable-diffusion-webui-gfx803-d682444ecc99319fbd2b142a12727501e2884ba7.tar.gz
stable-diffusion-webui-gfx803-d682444ecc99319fbd2b142a12727501e2884ba7.tar.bz2
stable-diffusion-webui-gfx803-d682444ecc99319fbd2b142a12727501e2884ba7.zip
add option to select hypernetwork modules when creating
Diffstat (limited to 'modules/ui.py')
-rw-r--r--modules/ui.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/ui.py b/modules/ui.py
index f2d16b12..14b87b92 100644
--- 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,