Fix bug in "remove last" button

This commit is contained in:
oobabooga 2023-01-22 13:10:36 -03:00
parent a410cf1345
commit c5cc3a3075

View File

@ -334,10 +334,11 @@ if args.chat or args.cai_chat:
def remove_last_message(name1, name2): def remove_last_message(name1, name2):
history.pop() history.pop()
_history = remove_example_dialogue_from_history(history)
if args.cai_chat: if args.cai_chat:
return generate_chat_html(history, name1, name2, character) return generate_chat_html(_history, name1, name2, character)
else: else:
return history return _history
def clear(): def clear():
global history global history