mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-12 23:01:41 +00:00
Add truncate_prompt_tokens param for vllm embedding
vllm embedding should allow sending "truncate_prompt_tokens". Links to https://github.com/vllm-project/vllm/blob/main/vllm/entrypoints/pooling/embed/protocol.py#L66
This commit is contained in:
parent
a5322c6efc
commit
c2e8da1e65
1 changed files with 1 additions and 0 deletions
|
|
@ -150,6 +150,7 @@ class HostedVLLMEmbeddingConfig(BaseEmbeddingConfig):
|
|||
"dimensions",
|
||||
"encoding_format",
|
||||
"user",
|
||||
"truncate_prompt_tokens",
|
||||
]
|
||||
|
||||
def map_openai_params(
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue