diff --git a/modules/chat.py b/modules/chat.py index d81d254..ce10e39 100644 --- a/modules/chat.py +++ b/modules/chat.py @@ -308,8 +308,8 @@ def generate_chat_reply_wrapper(text, state, regenerate=False, _continue=False): def remove_last_message(history): if len(history['visible']) > 0 and history['internal'][-1][0] != '<|BEGIN-VISIBLE-CHAT|>': - last = history['internal'].pop() - history['visible'].pop() + last = history['visible'].pop() + history['internal'].pop() else: last = ['', '']