mirror of
https://github.com/BerriAI/litellm.git
synced 2026-11-14 09:03:16 +00:00
Merge e020fab1e6 into 40be7fa1a8
This commit is contained in:
commit
6b36dabe68
1 changed files with 4 additions and 0 deletions
|
|
@ -114,6 +114,10 @@ class HuggingFaceRerankConfig(BaseRerankConfig):
|
|||
optional_rerank_params["texts"] = v
|
||||
elif k == "query" and v is not None:
|
||||
optional_rerank_params["query"] = v
|
||||
elif k == "truncate" and v is not None and isinstance(v, bool):
|
||||
optional_rerank_params["truncate"] = v
|
||||
elif k == "truncation_direction" and v is not None:
|
||||
optional_rerank_params["truncation_direction"] = v
|
||||
|
||||
return OptionalRerankParams(**optional_rerank_params)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue