OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-01-06 15:15:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

stylistic changes for #13118

Browse Source
This commit is contained in:
AUTOMATIC1111 2023-09-09 10:27:16 +03:00
parent 73c2a03d49
commit c9c457eda8
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
modules/ui.py
View File

@ -1,5 +1,4 @@
import datetime import datetime
import functools
import mimetypes import mimetypes
import os import os
import sys import sys
@ -152,7 +151,7 @@ def connect_clear_prompt(button):
) )
def update_token_counter(text, steps, is_positive=True): def update_token_counter(text, steps, *, is_positive=True):
try: try:
text, _ = extra_networks.parse_prompt(text) text, _ = extra_networks.parse_prompt(text)
@ -160,6 +159,7 @@ def update_token_counter(text, steps, is_positive=True):
_, prompt_flat_list, _ = prompt_parser.get_multicond_prompt_list([text]) _, prompt_flat_list, _ = prompt_parser.get_multicond_prompt_list([text])
else: else:
prompt_flat_list = [text] prompt_flat_list = [text]
prompt_schedules = prompt_parser.get_learned_conditioning_prompt_schedules(prompt_flat_list, steps) prompt_schedules = prompt_parser.get_learned_conditioning_prompt_schedules(prompt_flat_list, steps)
except Exception: except Exception:
@ -173,6 +173,10 @@ def update_token_counter(text, steps, is_positive=True):
return f"<span class='gr-box gr-text-input'>{token_count}/{max_length}</span>" return f"<span class='gr-box gr-text-input'>{token_count}/{max_length}</span>"
def update_negative_prompt_token_counter(text, steps):
return update_token_counter(text, steps, is_positive=False)
class Toprow: class Toprow:
"""Creates a top row UI with prompts, generate button, styles, extra little buttons for things, and enables some functionality related to their operation""" """Creates a top row UI with prompts, generate button, styles, extra little buttons for things, and enables some functionality related to their operation"""
@ -539,7 +543,7 @@ def create_ui():
] ]
toprow.token_button.click(fn=wrap_queued_call(update_token_counter), inputs=[toprow.prompt, steps], outputs=[toprow.token_counter]) toprow.token_button.click(fn=wrap_queued_call(update_token_counter), inputs=[toprow.prompt, steps], outputs=[toprow.token_counter])
toprow.negative_token_button.click(fn=wrap_queued_call(functools.partial(update_token_counter, is_positive=False)), inputs=[toprow.negative_prompt, steps], outputs=[toprow.negative_token_counter]) toprow.negative_token_button.click(fn=wrap_queued_call(update_negative_prompt_token_counter), inputs=[toprow.negative_prompt, steps], outputs=[toprow.negative_token_counter])
extra_networks_ui = ui_extra_networks.create_ui(txt2img_interface, [txt2img_generation_tab], 'txt2img') extra_networks_ui = ui_extra_networks.create_ui(txt2img_interface, [txt2img_generation_tab], 'txt2img')
ui_extra_networks.setup_ui(extra_networks_ui, txt2img_gallery) ui_extra_networks.setup_ui(extra_networks_ui, txt2img_gallery)