Skip to content

Commit

Permalink
Add model inf-retriever-v1 (#1744)
Browse files Browse the repository at this point in the history
* feat(models): add infly/inf-retriever-v1 model metadata- Add inf_models.py file with metadata for infly/inf-retriever-v1 model
- Update overview.py to include inf_models in model imports

* Reformat code

* Update inf-retriever-v1 ModelMeta

* Fill more information for inf-retriever-v1

* Add license information for inf-retriever-v1

---------

Co-authored-by: Samuel Yang <[email protected]>
  • Loading branch information
SamuelYang1 and SamuelYang1 authored Jan 15, 2025
1 parent 950f050 commit 60c4980
Show file tree
Hide file tree
Showing 2 changed files with 34 additions and 0 deletions.
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 @@ -62,6 +63,7 @@
gritlm_models,
gte_models,
ibm_granite_models,
inf_models,
jina_models,
linq_models,
llm2vec_models,
Expand Down

0 comments on commit 60c4980

Please sign in to comment.