Merge pull request #41 from DrJKL/patch-1

Handle spaces in STYLE function
asagi4 · May 2, 2024 · fc34ea5 · fc34ea5
2 parents 76dbe71 + 168aaf0
commit fc34ea5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/prompt_control/node_clip.py b/prompt_control/node_clip.py
@@ -189,6 +189,8 @@ def get_style(text, default_style="comfy", default_normalization="none"):
     if not styles:
         return default_style, default_normalization, text
     style, normalization = styles[0]
+    style = style.strip()
+    normalization = normalization.strip()
     if style not in AVAILABLE_STYLES:
         log.warning("Unrecognized prompt style: %s. Using %s", style, default_style)
         style = default_style