Jelajahi Sumber

Fix bug in loading chat history as text file

oobabooga 3 tahun lalu
induk
melakukan
8325e23923
1 mengubah file dengan 3 tambahan dan 0 penghapusan
  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():