Skip to content
32 changes: 32 additions & 0 deletions mteb/models/inf_models.py
Original file line number Diff line number Diff line change
@@ -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,
)
2 changes: 2 additions & 0 deletions mteb/models/overview.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
gritlm_models,
gte_models,
ibm_granite_models,
inf_models,
jasper_models,
jina_models,
linq_models,
Expand Down Expand Up @@ -61,6 +62,7 @@
gritlm_models,
gte_models,
ibm_granite_models,
inf_models,
jina_models,
linq_models,
llm2vec_models,
Expand Down
Loading