llama : add support for jina-reranker-v2 (#13900)

This commit is contained in:
Sigbjørn Skjæret 2025-05-29 21:42:31 +02:00 committed by GitHub
parent 2b131621e6
commit e83ba3e460
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 119 additions and 72 deletions

View file

@ -450,6 +450,7 @@ static const std::map<llm_arch, std::map<llm_tensor, const char *>> LLM_TENSOR_N
{ LLM_TENSOR_TOKEN_TYPES, "token_types" },
{ LLM_TENSOR_POS_EMBD, "position_embd" },
{ LLM_TENSOR_ATTN_OUT_NORM, "blk.%d.attn_output_norm" },
{ LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
{ LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
{ LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
{ LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },