diff --git a/mteb/models/inf_models.py b/mteb/models/inf_models.py new file mode 100644 index 000000000..4670b2073 --- /dev/null +++ b/mteb/models/inf_models.py @@ -0,0 +1,32 @@ +from __future__ import annotations + +from functools import partial + +from mteb.model_meta import ModelMeta, sentence_transformers_loader + +inf_retriever_v1 = ModelMeta( + loader=partial( # type: ignore + sentence_transformers_loader, + model_name="infly/inf-retriever-v1", + revision="d2d074546028c0012b5cc6af78c4fac24896e67f", + trust_remote_code=True, + ), + name="infly/inf-retriever-v1", + languages=["eng_Latn", "zho_Hans"], + open_weights=True, + revision="d2d074546028c0012b5cc6af78c4fac24896e67f", + release_date="2024-12-24", # initial commit of hf model. + n_parameters=7_069_121_024, + memory_usage=None, + embed_dim=3584, + license="apache-2.0", + max_tokens=131_072, + reference="https://huggingface.co/infly/inf-retriever-v1", + similarity_fn_name="cosine", + framework=["Sentence Transformers", "PyTorch"], + use_instructions=True, + adapted_from="Alibaba-NLP/gte-Qwen2-7B-instruct", + public_training_code=False, + public_training_data=False, + training_datasets=None, +) diff --git a/mteb/models/overview.py b/mteb/models/overview.py index 4e19bed19..868db8da4 100644 --- a/mteb/models/overview.py +++ b/mteb/models/overview.py @@ -23,6 +23,7 @@ gritlm_models, gte_models, ibm_granite_models, + inf_models, jasper_models, jina_models, linq_models, @@ -61,6 +62,7 @@ gritlm_models, gte_models, ibm_granite_models, + inf_models, jina_models, linq_models, llm2vec_models,