-
Notifications
You must be signed in to change notification settings - Fork 298
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* 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
1 parent
950f050
commit 60c4980
Showing
2 changed files
with
34 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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, | ||
) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters