Browse Source

Fix: Raptor: [Bug]: **ERROR**: Unknown field for GenerationConfig: max_tokens (#8331)

### What problem does this PR solve?

https://github.com/infiniflow/ragflow/issues/8324

### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.19.1
Stephen Hu 4 months ago
parent
commit
35034fed73
No account linked to committer's email address
1 changed files with 1 additions and 1 deletions
  1. 1
    1
      rag/llm/chat_model.py

+ 1
- 1
rag/llm/chat_model.py View File

@@ -1217,7 +1217,7 @@ class GeminiChat(Base):

def _clean_conf(self, gen_conf):
for k in list(gen_conf.keys()):
if k not in ["temperature", "top_p", "max_tokens"]:
if k not in ["temperature", "top_p"]:
del gen_conf[k]
return gen_conf


Loading…
Cancel
Save