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

Better handle spaces for Continue

Browse source
This commit is contained in:
oobabooga 2023-04-09 20:37:31 -03:00
parent b27d757fd1
commit 120f5662cf
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
modules/chat.py
View file

@ -171,8 +171,9 @@ def chatbot_wrapper(text, generate_state, name1, name2, context, mode, end_of_tu
shared.history['visible'].append(['', ''])
if _continue:
shared.history['internal'][-1] = [text, f'{last_reply[0]} {reply}']
shared.history['visible'][-1] = [visible_text, f'{last_reply[1]} {visible_reply}']
sep = list(map(lambda x : ' ' if x[-1] != ' ' else '', last_reply))
shared.history['internal'][-1] = [text, f'{last_reply[0]}{sep[0]}{reply}']
shared.history['visible'][-1] = [visible_text, f'{last_reply[1]}{sep[1]}{visible_reply}']
else:
shared.history['internal'][-1] = [text, reply]
shared.history['visible'][-1] = [visible_text, visible_reply]