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
5 changed files with 119 additions and 72 deletions

View File

@@ -1036,6 +1036,7 @@ MODEL_TENSORS: dict[MODEL_ARCH, list[MODEL_TENSOR]] = {
MODEL_TENSOR.POS_EMBD,
MODEL_TENSOR.OUTPUT_NORM,
MODEL_TENSOR.ATTN_OUT_NORM,
MODEL_TENSOR.ATTN_QKV,
MODEL_TENSOR.ATTN_Q,
MODEL_TENSOR.ATTN_K,
MODEL_TENSOR.ATTN_V,