apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix prompt segment order

This commit is contained in:
Sean Sube 2023-04-23 15:45:49 -05:00
parent bd3e6100a2
commit f1957f7378
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/diffusers/utils.py
View File

@ -40,14 +40,16 @@ def expand_alternative_ranges(prompt: str) -> List[str]:
next_group = ALTERNATIVE_RANGE.search(prompt)
while next_group is not None:
logger.debug("found alternative group in prompt: %s", next_group)
options = next_group.group().split("|")
prompt_groups.append(options)
if next_group.start() > (last_end + 1):
skipped_prompt = prompt[last_end:next_group.start()]
skipped_prompt = prompt[last_end + 1:next_group.start()]
logger.trace("appending skipped section of prompt: %s", skipped_prompt)
prompt_groups.append([skipped_prompt])
options = next_group.group()[1:-1].split("|")
logger.trace("split up alternative options: %s", options)
prompt_groups.append(options)
last_end = next_group.end()
next_group = ALTERNATIVE_RANGE.search(prompt, last_end)