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

Fix bug in loading chat history as text file

Browse source
This commit is contained in:
oobabooga 2023-01-23 14:28:02 -03:00
parent 059d47edb5
commit 8325e23923
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
server.py
View file

@ -366,6 +366,8 @@ if args.chat or args.cai_chat:
messages = [] messages = []
for i in range(len(idx)-1): for i in range(len(idx)-1):
messages.append(dialogue[idx[i]:idx[i+1]].strip()) messages.append(dialogue[idx[i]:idx[i+1]].strip())
messages.append(dialogue[idx[-1]:].strip())
history = [] history = []
entry = ['', ''] entry = ['', '']
for i in messages: for i in messages:
@ -376,6 +378,7 @@ if args.chat or args.cai_chat:
if not (len(entry[0]) == 0 and len(entry[1]) == 0): if not (len(entry[0]) == 0 and len(entry[1]) == 0):
history.append(entry) history.append(entry)
entry = ['', ''] entry = ['', '']
return history return history
def save_history(): def save_history():