Parcourir la source

fix various float parsing errors

AUTOMATIC il y a 2 ans
Parent
commit
f8e41a96bb
1 fichiers modifiés avec 1 ajouts et 2 suppressions
  1. 1 2
      modules/prompt_parser.py

+ 1 - 2
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 = []