simcop2387/text-generation-webui-mirror
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-06-08 14:46:14 -04:00
Code Issues Projects Releases Packages Wiki Activity

UI: Fix impersonate (#7025)

Browse source
This commit is contained in:
Underscore 2025-05-29 07:22:03 -04:00 • committed by GitHub
parent 75d6cfd14d
commit 63234b9b6f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
modules/chat.py
View file

@ -691,16 +691,19 @@ def chatbot_wrapper(text, state, regenerate=False, _continue=False, loading_mess
yield output yield output
def impersonate_wrapper(text, state): def impersonate_wrapper(textbox, state):
text = textbox['text']
static_output = chat_html_wrapper(state['history'], state['name1'], state['name2'], state['mode'], state['chat_style'], state['character_menu']) static_output = chat_html_wrapper(state['history'], state['name1'], state['name2'], state['mode'], state['chat_style'], state['character_menu'])
prompt = generate_chat_prompt('', state, impersonate=True) prompt = generate_chat_prompt('', state, impersonate=True)
stopping_strings = get_stopping_strings(state) stopping_strings = get_stopping_strings(state)
yield text + '...', static_output textbox['text'] = text + '...'
yield textbox, static_output
reply = None reply = None
for reply in generate_reply(prompt + text, state, stopping_strings=stopping_strings, is_chat=True): for reply in generate_reply(prompt + text, state, stopping_strings=stopping_strings, is_chat=True):
yield (text + reply).lstrip(' '), static_output textbox['text'] = (text + reply).lstrip(' ')
yield textbox, static_output
if shared.stop_everything: if shared.stop_everything:
return return