|
@@ -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():
|
oobabooga