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

Update impersonate_wrapper

Browse source
This commit is contained in:
oobabooga 2023-05-12 12:59:48 -03:00
parent e283ddc559
commit df37ba5256
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
modules/chat.py
View file

@ -247,10 +247,12 @@ def impersonate_wrapper(text, state):
if next_character_found: if next_character_found:
break break
if reply is not None: if reply in [None, '']:
break
else:
cumulative_reply = reply cumulative_reply = reply
yield reply yield cumulative_reply
def generate_chat_reply(text, state, regenerate=False, _continue=False): def generate_chat_reply(text, state, regenerate=False, _continue=False):