Update impersonate_wrapper

This commit is contained in:
oobabooga 2023-05-12 12:59:48 -03:00
parent e283ddc559
commit df37ba5256

View file

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