simcop2387/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

Fix impersonate when some text is present (closes #3564)

Browse source
This commit is contained in:
oobabooga 2023-08-13 21:10:47 -07:00
parent cc7e6ef645
commit ff9b5861c8
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
modules/chat.py
View file

@ -272,8 +272,8 @@ def impersonate_wrapper(text, start_with, state):
yield text + '...' yield text + '...'
reply = None reply = None
for reply in generate_reply(prompt, state, stopping_strings=stopping_strings, is_chat=True): for reply in generate_reply(prompt + text, state, stopping_strings=stopping_strings, is_chat=True):
yield reply.lstrip(' ') yield (text + reply).lstrip(' ')
if shared.stop_everything: if shared.stop_everything:
return return