fix: fix jina v1, 2 models (#1872)

fix jina models
embeddings-benchmark · Jan 26, 2025 · 1d66089 · 1d66089
1 parent 95714d0
commit 1d66089
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 8 deletions.
diff --git a/mteb/models/gme_models.py b/mteb/models/gme_models.py
@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import logging
-from functools import partial
 
 from mteb.model_meta import ModelMeta
 

diff --git a/mteb/models/jina_models.py b/mteb/models/jina_models.py
@@ -245,6 +245,12 @@ def encode(
 
 
 jina_embeddings_v2_base_en = ModelMeta(
+    loader=partial(
+        SentenceTransformerWrapper,
+        model_name="jinaai/jina-embeddings-v2-base-en",
+        revision="6e85f575bc273f1fd840a658067d0157933c83f0",
+        trust_remote_code=True,
+    ),
     name="jinaai/jina-embeddings-v2-base-en",
     languages=["eng-Latn"],
     open_weights=True,
@@ -266,6 +272,12 @@ def encode(
 )
 
 jina_embeddings_v2_small_en = ModelMeta(
+    loader=partial(
+        SentenceTransformerWrapper,
+        model_name="jinaai/jina-embeddings-v2-small-en",
+        revision="796cff318cdd4e5fbe8b7303a1ef8cbec36996ef",
+        trust_remote_code=True,
+    ),
     name="jinaai/jina-embeddings-v2-small-en",
     languages=["eng-Latn"],
     open_weights=True,
@@ -287,6 +299,12 @@ def encode(
 )
 
 jina_embedding_b_en_v1 = ModelMeta(
+    loader=partial(
+        SentenceTransformerWrapper,
+        model_name="jinaai/jina-embedding-b-en-v1",
+        revision="aa0645035294a8c0607ce5bb700aba982cdff32c",
+        trust_remote_code=True,
+    ),
     name="jinaai/jina-embedding-b-en-v1",
     languages=["eng-Latn"],
     open_weights=True,
@@ -308,6 +326,12 @@ def encode(
 )
 
 jina_embedding_s_en_v1 = ModelMeta(
+    loader=partial(
+        SentenceTransformerWrapper,
+        model_name="jinaai/jina-embedding-s-en-v1",
+        revision="c1fed70aa4823a640f1a7150a276e4d3b08dce08",
+        trust_remote_code=True,
+    ),
     name="jinaai/jina-embedding-s-en-v1",
     languages=["eng-Latn"],
     open_weights=True,

diff --git a/mteb/models/lens_models.py b/mteb/models/lens_models.py
@@ -1,12 +1,6 @@
 from __future__ import annotations
 
-from functools import partial
-
-import torch
-
-from mteb.encoder_interface import PromptType
-from mteb.model_meta import ModelMeta, sentence_transformers_loader
-from mteb.models.instruct_wrapper import instruct_wrapper
+from mteb.model_meta import ModelMeta
 
 from .bge_models import bge_full_data