Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

update parser

Browse Source
This commit is contained in:
Vladimir Mandic
2023-05-18 19:46:11 -04:00
parent 1c70056744
commit 527dc0eedf
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/prompt_parser.py
+3 -1
View File
@@ -317,6 +317,7 @@ def parse_prompt_attention(text):
for m in re_attention.finditer(text):
text = m.group(0)
weight = m.group(1)
if text.startswith('\\'):
res.append([text[1:], 1.0])
elif text == '(':
@@ -326,7 +327,8 @@ def parse_prompt_attention(text):
elif weight is not None and len(round_brackets) > 0:
multiply_range(round_brackets.pop(), float(weight))
elif weight is not None and len(square_brackets) > 0:
multiply_range(square_brackets.pop(), float(weight))
if opts.prompt_attention == 'Full parser':
multiply_range(square_brackets.pop(), float(weight))
elif text == ')' and len(round_brackets) > 0:
multiply_range(round_brackets.pop(), round_bracket_multiplier)
elif text == ']' and len(square_brackets) > 0: