From e3d2b548d8df716bd5ab8ef4f080d7cff82d51cf Mon Sep 17 00:00:00 2001 From: Kenneth Enevoldsen Date: Fri, 29 Nov 2024 08:35:51 +0100 Subject: [PATCH] fix: Ensure that models match the names on embedding-benchmarks/results (#1519) --- mteb/models/openai_models.py | 6 +++--- mteb/models/voyage_models.py | 12 ++++++------ 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/mteb/models/openai_models.py b/mteb/models/openai_models.py index 50967e898b..bb8550b79a 100644 --- 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 index ea6b25bde1..3cf8b286e1 100644 --- 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