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": {
diff --git a/scripts/forge_hidiffusion.py b/scripts/forge_hidiffusion.py
index 0b0784b..45e97e6 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"):
@@ -68,10 +67,17 @@ def ui(self, *args, **kwargs):
with InputAccordion(False, label="Advanced Options") as use_raunet_advanced:
with gr.Group():
+ gr.HTML(
+ """
+ Recommended block settings:
+