From 975372fef498e6797e8a6a38af974391793a952c Mon Sep 17 00:00:00 2001
From: fzoll <fodizoltan@gmail.com>
Date: Sun, 24 Aug 2025 11:44:51 +0200
Subject: [PATCH] Correcting the JINA models with SentenceTransformerWrapper

---
 mteb/models/jina_models.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/mteb/models/jina_models.py b/mteb/models/jina_models.py
index 6aef19fd5a..d63522f1af 100644
--- a/mteb/models/jina_models.py
+++ b/mteb/models/jina_models.py
@@ -633,7 +633,7 @@ def get_programming_task_override(
 jina_embeddings_v2_base_en = ModelMeta(
     loader=partial(
         SentenceTransformerWrapper,
-        model_name="jinaai/jina-embeddings-v2-base-en",
+        model="jinaai/jina-embeddings-v2-base-en",
         revision="6e85f575bc273f1fd840a658067d0157933c83f0",
         trust_remote_code=True,
     ),
@@ -692,7 +692,7 @@ def get_programming_task_override(
 jina_embeddings_v2_small_en = ModelMeta(
     loader=partial(
         SentenceTransformerWrapper,
-        model_name="jinaai/jina-embeddings-v2-small-en",
+        model="jinaai/jina-embeddings-v2-small-en",
         revision="44e7d1d6caec8c883c2d4b207588504d519788d0",
         trust_remote_code=True,
     ),
@@ -751,7 +751,7 @@ def get_programming_task_override(
 jina_embedding_b_en_v1 = ModelMeta(
     loader=partial(
         SentenceTransformerWrapper,
-        model_name="jinaai/jina-embedding-b-en-v1",
+        model="jinaai/jina-embedding-b-en-v1",
         revision="32aa658e5ceb90793454d22a57d8e3a14e699516",
     ),
     name="jinaai/jina-embedding-b-en-v1",
@@ -805,7 +805,7 @@ def get_programming_task_override(
 jina_embedding_s_en_v1 = ModelMeta(
     loader=partial(
         SentenceTransformerWrapper,
-        model_name="jinaai/jina-embedding-s-en-v1",
+        model="jinaai/jina-embedding-s-en-v1",
         revision="5ac6cd473e2324c6d5f9e558a6a9f65abb57143e",
     ),
     name="jinaai/jina-embedding-s-en-v1",