Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-06 21:10:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

add setting for prompt linebreaks

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-02-01 09:41:52 -05:00
parent 64a5ca6471
commit 8e03755d4b
4 changed files with 27 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/prompt_parser_diffusers.py
+18 -3
View File
@@ -354,13 +354,28 @@ def get_prompts_with_weights(pipe, prompt: str):
if shared.opts.prompt_mean_norm:
texts_and_weights = normalize_prompt(texts_and_weights)
texts, text_weights = zip(*texts_and_weights)
if debug_enabled:
avg_weight = 0
min_weight = 1
max_weight = 0
sections = 0
try:
all_tokens = 0
for text in texts:
for text, weight in zip(texts, text_weights):
tokens = get_tokens(pipe, 'section', text)
all_tokens += tokens
debug(f'Prompt tokenizer: parser={shared.opts.prompt_attention} tokens={all_tokens}')
avg_weight += tokens*weight
min_weight = min(min_weight, weight)
max_weight = max(max_weight, weight)
if text != 'BREAK':
sections += 1
if all_tokens > 0:
avg_weight = avg_weight / all_tokens
shared.log.debug(f'Prompt tokenizer: parser={shared.opts.prompt_attention} len={len(prompt)} sections={sections} tokens={all_tokens} weights={min_weight:.2f}/{avg_weight:.2f}/{max_weight:.2f}')
except Exception:
pass
debug(f'Prompt: weights={texts_and_weights} time={(time.time() - t0):.3f}')
return texts, text_weights