fix: Ensure that models match the names on embedding-benchmarks/resul…

…ts (#1519)
embeddings-benchmark · Nov 29, 2024 · e3d2b54 · e3d2b54
1 parent 377a63d
commit e3d2b54
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/mteb/models/openai_models.py b/mteb/models/openai_models.py
@@ -56,7 +56,7 @@ def _to_numpy(self, embedding_response) -> np.ndarray:
 
 
 text_embedding_3_small = ModelMeta(
-    name="text-embedding-3-small",
+    name="openai/text-embedding-3-small",
     revision="1",
     release_date="2024-01-25",
     languages=None,  # supported languages not specified
@@ -73,7 +73,7 @@ def _to_numpy(self, embedding_response) -> np.ndarray:
     use_instructions=False,
 )
 text_embedding_3_large = ModelMeta(
-    name="text-embedding-3-large",
+    name="openai/text-embedding-3-large",
     revision="1",
     release_date="2024-01-25",
     languages=None,  # supported languages not specified
@@ -87,7 +87,7 @@ def _to_numpy(self, embedding_response) -> np.ndarray:
     memory_usage=None,
 )
 text_embedding_ada_002 = ModelMeta(
-    name="text-embedding-ada-002",
+    name="openai/text-embedding-ada-002",
     revision="1",
     release_date="2022-12-15",
     languages=None,  # supported languages not specified

diff --git a/mteb/models/voyage_models.py b/mteb/models/voyage_models.py
@@ -141,7 +141,7 @@ def _batched_encode(
 }
 
 voyage_large_2_instruct = ModelMeta(
-    name="voyage-large-2-instruct",
+    name="voyageai/voyage-large-2-instruct",
     revision="1",
     release_date="2024-05-05",
     languages=None,  # supported languages not specified
@@ -163,7 +163,7 @@ def _batched_encode(
 )
 
 voyage_finance_2 = ModelMeta(
-    name="voyage-finance-2",
+    name="voyageai/voyage-finance-2",
     revision="1",
     release_date="2024-05-30",
     languages=None,  # supported languages not specified
@@ -185,7 +185,7 @@ def _batched_encode(
 )
 
 voyage_law_2 = ModelMeta(
-    name="voyage-law-2",
+    name="voyageai/voyage-law-2",
     revision="1",
     release_date="2024-04-15",
     languages=None,  # supported languages not specified
@@ -207,7 +207,7 @@ def _batched_encode(
 )
 
 voyage_code_2 = ModelMeta(
-    name="voyage-code-2",
+    name="voyageai/voyage-code-2",
     revision="1",
     release_date="2024-01-23",
     languages=None,  # supported languages not specified
@@ -251,7 +251,7 @@ def _batched_encode(
 )
 
 voyage_2 = ModelMeta(
-    name="voyage-2",
+    name="voyageai/voyage-2",
     revision="1",
     release_date="2023-10-29",
     languages=None,  # supported languages not specified
@@ -272,7 +272,7 @@ def _batched_encode(
     use_instructions=False,
 )
 voyage_multilingual_2 = ModelMeta(
-    name="voyage-multilingual-2",
+    name="voyageai/voyage-multilingual-2",
     revision="1",
     release_date="2024-06-10",
     languages=None,  # supported languages not specified