github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-11 13:59:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

[colossalqa] fix pangu api (#5170)

Browse Source 
* fix pangu api

* add comment
This commit is contained in:
Michelle
2023-12-11 14:08:11 +08:00
committed by GitHub
parent 21aa5de00b
commit b07a6f4e27
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
applications/ColossalQA/colossalqa/memory.py
View File

@@ -154,7 +154,7 @@ class ConversationBufferWithSummary(ConversationSummaryMemory):
remain = self.max_tokens - prompt_length
while self.get_conversation_length() > remain:
if len(self.buffered_history.messages) <= 2:
raise RuntimeError("Exeeed max_tokens, trunck size of retrieved documents is too large")
raise RuntimeError("Exceed max_tokens, trunk size of retrieved documents is too large")
temp = self.buffered_history.messages.pop(0)
self.summarized_history_temp.messages.append(temp)
temp = self.buffered_history.messages.pop(0)