workaround for a mysterious bug where prompt weights can't be matched

3 years ago · 2d3ea42a2d
parent 5f24b7bcf4
commit 2d3ea42a2d
1 changed files with 3 additions and 1 deletions
--- a/modules/prompt_parser.py
+++ b/modules/prompt_parser.py
@ -156,7 +156,9 @@ def get_multicond_prompt_list(prompts):

        indexes = []
        for subprompt in subprompts:
-            text, weight = re_weight.search(subprompt).groups()
+            match = re_weight.search(subprompt)
+
+            text, weight = match.groups() if match is not None else (subprompt, 1.0)

            weight = float(weight) if weight is not None else 1.0