From 2dcfacab469f4e6a5727529fdefd8dc06e4c3664 Mon Sep 17 00:00:00 2001 From: Brendan Lackey Date: Thu, 21 Nov 2024 22:07:46 -0700 Subject: [PATCH 1/3] adjust info text --- scripts/forge_hidiffusion.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/scripts/forge_hidiffusion.py b/scripts/forge_hidiffusion.py index 0b0784b..983364b 100644 --- a/scripts/forge_hidiffusion.py +++ b/scripts/forge_hidiffusion.py @@ -36,10 +36,9 @@ def show(self, is_img2img): def ui(self, *args, **kwargs): with InputAccordion(False, label=self.title()) as enabled: model_type = gr.Radio( - choices=["SD15", "SDXL"], + choices=["SD 1.5/2.1", "SDXL"], value=lambda: "SDXL", label="Model Type", - info="Note: Use SD15 setting for SD 2.1 as well.", ) with gr.Tab("RAUNet"): @@ -71,7 +70,7 @@ def ui(self, *args, **kwargs): raunet_input_blocks = gr.Text(label="Input Blocks", value="3") raunet_output_blocks = gr.Text(label="Output Blocks", value="8") gr.Markdown( - "For SD1.5: Input 3 corresponds to Output 8, Input 6 to Output 5, Input 9 to Output 2" + "For SD1.5/2.1: Input 3 corresponds to Output 8, Input 6 to Output 5, Input 9 to Output 2" ) gr.Markdown("For SDXL: Input 3 corresponds to Output 5, Input 6 to Output 2") @@ -127,7 +126,7 @@ def ui(self, *args, **kwargs): gr.Markdown("Advanced MSW-MSA settings. For fine-tuning performance and quality improvements.") with gr.Group(): gr.HTML( - "Recommended block settings:
" + "Recommended block settings:
" ) mswmsa_input_blocks = gr.Text(label="Input Blocks", value="1,2") mswmsa_middle_blocks = gr.Text(label="Middle Blocks", value="") @@ -158,7 +157,7 @@ def ui(self, *args, **kwargs): # Add JavaScript to handle visibility and model-specific settings def update_raunet_settings(model_type): - if model_type == "SD15": + if model_type == "SD 1.5/2.1": return "3", "8", "4", "8", 0.0, 0.45, 0.0, 0.3 else: # SDXL return ( @@ -188,7 +187,7 @@ def update_raunet_settings(model_type): ) def update_mswmsa_settings(model_type): - if model_type == "SD15": + if model_type == "SD 1.5/2.1": return "1,2", "", "9,10,11" else: # SDXL return "4,5", "", "4,5" From d3585624ca74cfc466b0d597ba2767aa920d559f Mon Sep 17 00:00:00 2001 From: Brendan Lackey Date: Thu, 21 Nov 2024 22:10:13 -0700 Subject: [PATCH 2/3] adjust info text --- scripts/forge_hidiffusion.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/scripts/forge_hidiffusion.py b/scripts/forge_hidiffusion.py index 983364b..45e97e6 100644 --- a/scripts/forge_hidiffusion.py +++ b/scripts/forge_hidiffusion.py @@ -67,6 +67,13 @@ def ui(self, *args, **kwargs): with InputAccordion(False, label="Advanced Options") as use_raunet_advanced: with gr.Group(): + gr.HTML( + """ + Recommended block settings:
+ + """ + ) raunet_input_blocks = gr.Text(label="Input Blocks", value="3") raunet_output_blocks = gr.Text(label="Output Blocks", value="8") gr.Markdown( @@ -126,7 +133,11 @@ def ui(self, *args, **kwargs): gr.Markdown("Advanced MSW-MSA settings. For fine-tuning performance and quality improvements.") with gr.Group(): gr.HTML( - "Recommended block settings:
" + """ + Recommended block settings:
+ + """ ) mswmsa_input_blocks = gr.Text(label="Input Blocks", value="1,2") mswmsa_middle_blocks = gr.Text(label="Middle Blocks", value="") From 42568caabfd24d5c446dc6c589c3410cba09eeb6 Mon Sep 17 00:00:00 2001 From: Brendan Lackey Date: Thu, 21 Nov 2024 22:11:14 -0700 Subject: [PATCH 3/3] adjust info text --- hidiffusion/raunet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hidiffusion/raunet.py b/hidiffusion/raunet.py index 5cf04da..219cb98 100644 --- a/hidiffusion/raunet.py +++ b/hidiffusion/raunet.py @@ -369,7 +369,7 @@ def configure_blocks( enabled = True model_configs = { - "SD15": { + "SD 1.5/2.1": { "blocks": ("3", "8"), "ca_blocks": ("1", "11"), "modes": {