diff --git a/mteb/models/overview.py b/mteb/models/overview.py index 399d8359c..101e00043 100644 --- a/mteb/models/overview.py +++ b/mteb/models/overview.py @@ -48,6 +48,7 @@ ru_sentence_models, salesforce_models, sentence_transformers_models, + spice_models, stella_models, text2vec_models, uae_models, @@ -72,6 +73,7 @@ gme_models, ibm_granite_models, inf_models, + spice_models, jina_models, lens_models, linq_models, diff --git a/mteb/models/spice_models.py b/mteb/models/spice_models.py new file mode 100644 index 000000000..0f471e4f9 --- /dev/null +++ b/mteb/models/spice_models.py @@ -0,0 +1,32 @@ +from __future__ import annotations + +from functools import partial + +from mteb.model_meta import ModelMeta, sentence_transformers_loader + +model_prompts = {"query": "Represent this sentence for searching relevant passages: "} + +spice = ModelMeta( + loader=partial( # type: ignore + sentence_transformers_loader, + model_name="iamgroot42/spice", + revision="1d0fdb2b5d7aed81bba960cd8c85671674d49bdc", + model_prompts=model_prompts, + ), + name="iamgroot42/spice", + languages=["eng_Latn"], + open_weights=True, + revision="1d0fdb2b5d7aed81bba960cd8c85671674d49bdc", + release_date="2025-01-27", # initial commit of hf model. + n_parameters=24_000_000, + embed_dim=512, + license="mit", + max_tokens=512, + reference="https://huggingface.co/iamgroot42/spice", + similarity_fn_name="cosine", + framework=["Sentence Transformers", "PyTorch"], + use_instructions=True, + public_training_code=None, + public_training_data=None, + training_datasets=None, +)