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

@ -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,