mirror of
https://github.com/StanGirard/quivr.git
synced 2025-01-05 18:57:48 +03:00
Add max_input and max_tokens parameters to KnowledgeBrainQA constructor (#2247)
This pull request adds the `max_input` and `max_tokens` parameters to the `KnowledgeBrainQA` constructor. These parameters allow for setting the maximum input length and maximum number of tokens for the KnowledgeBrainQA model. This provides more flexibility and control over the input size and tokenization process.
This commit is contained in:
parent
81072b3841
commit
e4f920120f
@ -181,6 +181,8 @@ class KnowledgeBrainQA(BaseModel, QAInterface):
|
||||
brain_id=brain_id,
|
||||
chat_id=chat_id,
|
||||
streaming=streaming,
|
||||
max_input=self.max_input,
|
||||
max_tokens=self.max_tokens,
|
||||
**kwargs,
|
||||
)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user