1580158
/
sd_extension-prompt_formatter
mirror of https://github.com/uwidev/sd_extension-prompt_formatter.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: add check for BRACKET2WEIGHT 

Accidentally removed when refactoring.

Fixes #21.
main v0.5.1
uwidev 2024-09-30 09:35:29 -07:00
parent 5920bef70b
commit 4716fe1149
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/prompt_formatter.py
View File

@ -43,7 +43,7 @@ def format_prompt(*prompts: tuple[dict]):
prompt = pipeline.align_colons(prompt)
prompt = pipeline.align_commas(prompt, do_it=SPACE_COMMAS)
prompt = pipeline.align_alternating(prompt)
prompt = pipeline.bracket_to_weights(prompt)
prompt = pipeline.bracket_to_weights(prompt, do_it=BRACKET2WEIGHT)
ret.append(prompt.strip())

4
tests/test_pipeline.py
View File

@ -124,5 +124,7 @@ def test_space_to_underscore():
assert pipeline.space_to_underscore('one two three') == 'one_two_three'
assert pipeline.space_to_underscore('this is a test') == 'this_is_a_test'
assert pipeline.space_to_underscore('<embed:foo bar>, baz') == '<embed:foo bar>, baz'
assert pipeline.space_to_underscore('some_var_name', opposite=False) == 'some var name'
pipeline.BRACKET2WEIGHT = False
assert pipeline.space_to_underscore('some_var_name', opposite=pipeline.BRACKET2WEIGHT) == 'some var name'