fix various float parsing errors

3 years ago · f8e41a96bb
parent c26732fbee
commit f8e41a96bb
1 changed files with 1 additions and 2 deletions
--- a/modules/prompt_parser.py
+++ b/modules/prompt_parser.py
@ -143,8 +143,7 @@ def get_learned_conditioning(model, prompts, steps):


 re_AND = re.compile(r"\bAND\b")
-re_weight = re.compile(r"^(.*?)(?:\s*:\s*([-+]?\s*(?:\d+|\d*\.\d+)?))?\s*$")
-
+re_weight = re.compile(r"^(.*?)(?:\s*:\s*([-+]?(?:\d+\.?|\d*\.\d+)))?\s*$")

 def get_multicond_prompt_list(prompts):
    res_indexes = []