File size: 466 Bytes
750b88d | 1 2 3 4 5 6 7 8 9 10 11 12 | --- a/HippoRAG/src/hipporag/rerank.py
+++ b/HippoRAG/src/hipporag/rerank.py
@@ -90,7 +90,7 @@
messages.append({"role": "user", "content": self.one_input_template.format(question=question, fact_before_filter=fact_before_filter)})
# call openai
- self.default_gen_kwargs['max_completion_tokens'] = 512
+ self.default_gen_kwargs['max_completion_tokens'] = 10000
response = self.llm_infer_fn(
messages=messages,
|