Explorar el Código

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 hace 4 meses
padre
commit
35034fed73
No account linked to committer's email address
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. 1
    1
      rag/llm/chat_model.py

+ 1
- 1
rag/llm/chat_model.py Ver fichero



def _clean_conf(self, gen_conf): def _clean_conf(self, gen_conf):
for k in list(gen_conf.keys()): 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] del gen_conf[k]
return gen_conf return gen_conf



Cargando…
Cancelar
Guardar