소스 검색

kimi test

郑毅 3 달 전
부모
커밋
74560db0d8
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      kimi/chat.py

+ 1 - 1
kimi/chat.py

@@ -36,7 +36,7 @@ class Chat:
             "content": result
         })
         # 随着对话的进行,模型每次需要传入的 token 都会线性增加,只保留特定几轮对话
-        if self.save_first_len + self.save_last_len > len(self.history):
+        if self.save_first_len + self.save_last_len < len(self.history):
             self.history = self.history[0:self.save_first_len] + self.history[len(self.history) - self.save_last_len:]
         
         return result