Преглед на файлове

Fix bug in loading chat history as text file

oobabooga преди 3 години
родител
ревизия
8325e23923
променени са 1 файла, в които са добавени 3 реда и са изтрити 0 реда
  1. 3 0
      server.py

+ 3 - 0
server.py

@@ -366,6 +366,8 @@ if args.chat or args.cai_chat:
         messages = []
         for i in range(len(idx)-1):
             messages.append(dialogue[idx[i]:idx[i+1]].strip())
+        messages.append(dialogue[idx[-1]:].strip())
+
         history = []
         entry = ['', '']
         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):
                     history.append(entry)
                 entry = ['', '']
+
         return history
 
     def save_history():