From 97643199d8b9cd53a5bc58225464748d1e1be892 Mon Sep 17 00:00:00 2001 From: vaibhavad Date: Fri, 14 Jun 2024 19:22:05 +0000 Subject: [PATCH 1/2] add gritlm to init file --- mteb/models/__init__.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/mteb/models/__init__.py b/mteb/models/__init__.py index 6a4010457f..62bd1e7418 100644 --- a/mteb/models/__init__.py +++ b/mteb/models/__init__.py @@ -9,6 +9,7 @@ from mteb.model_meta import ModelMeta from mteb.models import ( e5_models, + gritlm, openai_models, sentence_transformers_models, voyage_models, @@ -103,7 +104,13 @@ def model_meta_from_sentence_transformers(model: SentenceTransformer) -> ModelMe return meta -model_modules = [e5_models, sentence_transformers_models, openai_models, voyage_models] +model_modules = [ + e5_models, + sentence_transformers_models, + openai_models, + voyage_models, + gritlm, +] models = {} From a5924002ed2f448507aad0b9d20e89504519a45e Mon Sep 17 00:00:00 2001 From: Niklas Muennighoff Date: Fri, 14 Jun 2024 12:25:59 -0700 Subject: [PATCH 2/2] sort --- mteb/models/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mteb/models/__init__.py b/mteb/models/__init__.py index 62bd1e7418..b3d8717a4b 100644 --- a/mteb/models/__init__.py +++ b/mteb/models/__init__.py @@ -106,10 +106,10 @@ def model_meta_from_sentence_transformers(model: SentenceTransformer) -> ModelMe model_modules = [ e5_models, - sentence_transformers_models, + gritlm, openai_models, + sentence_transformers_models, voyage_models, - gritlm, ] models = {}