hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 03:10:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add before_token_counter callback and use it for prompt comments

Browse Source
This commit is contained in:
AUTOMATIC1111
2024-02-11 12:56:53 +03:00
parent 02ab75b86a
commit b7f45e67dc
3 changed files with 43 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modules/ui.py
View File

@@ -152,6 +152,12 @@ def connect_clear_prompt(button):
def update_token_counter(text, steps, styles, *, is_positive=True):
params = script_callbacks.BeforeTokenCounterParams(text, steps, styles, is_positive=is_positive)
script_callbacks.before_token_counter_callback(params)
text = params.prompt
steps = params.steps
styles = params.styles
is_positive = params.is_positive
if shared.opts.include_styles_into_token_counters:
apply_styles = shared.prompt_styles.apply_styles_to_prompt if is_positive else shared.prompt_styles.apply_negative_styles_to_prompt