mteb 2.5.2__py3-none-any.whl → 2.7.2__py3-none-any.whl

mteb/models/model_implementations/mixedbread_ai_models.py

@@ -0,0 +1,332 @@
+from mteb.models.model_implementations.pylate_models import MultiVectorModel
+from mteb.models.model_meta import (
+    ModelMeta,
+    ScoringFunction,
+)
+from mteb.models.sentence_transformer_wrapper import (
+    CrossEncoderWrapper,
+    sentence_transformers_loader,
+)
+
+mixedbread_training_data = {
+    # from correspondence:
+    # as mentioned in our blog post
+    # (https://www.mixedbread.com/blog/mxbai-embed-large-v1#built-for-rag-and-real-world-use-cases:~:text=During%20the%20whole,related%20use%20cases.)
+    # We do not train on any data (except the MSMarco training split) of MTEB. We have a strong filtering process to ensure the OOD setting. That's true
+    # for all of our models. Keep up the good work and let me know if you have any questions.
+    "MSMARCO",
+}
+
+mxbai_embed_large_v1 = ModelMeta(
+    loader=sentence_transformers_loader,
+    loader_kwargs=dict(
+        model_prompts={
+            "query": "Represent this sentence for searching relevant passages: "
+        },
+    ),
+    name="mixedbread-ai/mxbai-embed-large-v1",
+    model_type=["dense"],
+    languages=["eng-Latn"],
+    open_weights=True,
+    revision="990580e27d329c7408b3741ecff85876e128e203",
+    release_date="2024-03-07",  # initial commit of hf model.
+    n_parameters=335_000_000,
+    memory_usage_mb=639,
+    max_tokens=512,
+    embed_dim=1024,
+    license="apache-2.0",
+    reference="https://huggingface.co/mixedbread-ai/mxbai-embed-large-v1",
+    similarity_fn_name=ScoringFunction.COSINE,
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "ONNX",
+        "safetensors",
+        "GGUF",
+        "Transformers",
+    ],
+    use_instructions=True,
+    citation="""
+    @online{emb2024mxbai,
+      title={Open Source Strikes Bread - New Fluffy Embeddings Model},
+      author={Sean Lee and Aamir Shakir and Darius Koenig and Julius Lipp},
+      year={2024},
+      url={https://www.mixedbread.ai/blog/mxbai-embed-large-v1},
+    }
+
+    @article{li2023angle,
+      title={AnglE-optimized Text Embeddings},
+      author={Li, Xianming and Li, Jing},
+      journal={arXiv preprint arXiv:2309.12871},
+      year={2023}
+    }
+    """,
+    public_training_code=None,
+    public_training_data=None,
+    training_datasets=mixedbread_training_data,
+)
+
+mxbai_embed_2d_large_v1 = ModelMeta(
+    loader=sentence_transformers_loader,
+    name="mixedbread-ai/mxbai-embed-2d-large-v1",
+    model_type=["dense"],
+    languages=["eng-Latn"],
+    open_weights=True,
+    revision="7e639ca8e344af398876ead3b19ec3c0b9068f49",
+    release_date="2024-03-04",  # initial commit of hf model.
+    n_parameters=335_000_000,
+    memory_usage_mb=None,
+    max_tokens=512,
+    embed_dim=768,
+    license="apache-2.0",
+    reference="https://huggingface.co/mixedbread-ai/mxbai-embed-2d-large-v1",
+    similarity_fn_name=ScoringFunction.COSINE,
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "ONNX",
+        "safetensors",
+        "Transformers",
+    ],
+    use_instructions=True,
+    adapted_from=None,
+    superseded_by=None,
+    public_training_code=None,
+    public_training_data=None,
+    training_datasets=None,
+)
+
+
+mxbai_embed_xsmall_v1 = ModelMeta(
+    loader=sentence_transformers_loader,
+    name="mixedbread-ai/mxbai-embed-xsmall-v1",
+    model_type=["dense"],
+    languages=["eng-Latn"],
+    open_weights=True,
+    revision="2f741ec33328bb57e4704e1238fc59a4a5745705",
+    release_date="2024-08-13",  # initial commit of hf model.
+    n_parameters=24_100_000,
+    memory_usage_mb=None,
+    max_tokens=512,
+    embed_dim=384,
+    license="apache-2.0",
+    reference="https://huggingface.co/mixedbread-ai/mxbai-embed-xsmall-v1",
+    similarity_fn_name=ScoringFunction.COSINE,
+    framework=["Sentence Transformers", "PyTorch", "ONNX", "safetensors", "GGUF"],
+    use_instructions=True,
+    adapted_from="sentence-transformers/all-MiniLM-L6-v2",
+    superseded_by=None,
+    public_training_code=None,
+    public_training_data=None,
+    training_datasets=mixedbread_training_data,
+    citation="""@online{xsmall2024mxbai,
+  title={Every Byte Matters: Introducing mxbai-embed-xsmall-v1},
+  author={Sean Lee and Julius Lipp and Rui Huang and Darius Koenig},
+  year={2024},
+  url={https://www.mixedbread.ai/blog/mxbai-embed-xsmall-v1},
+}""",
+)
+
+mxbai_rerank_xsmall_v1 = ModelMeta(
+    loader=CrossEncoderWrapper,
+    name="mixedbread-ai/mxbai-rerank-xsmall-v1",
+    revision="b5c6e9da73abc3711f593f705371cdbe9e0fe422",
+    release_date="2024-02-29",
+    languages=["eng-Latn"],
+    n_parameters=70830337,
+    memory_usage_mb=135.0,
+    max_tokens=512,
+    embed_dim=None,
+    license="apache-2.0",
+    open_weights=True,
+    public_training_code=None,
+    public_training_data=None,
+    framework=[
+        "PyTorch",
+        "Sentence Transformers",
+        "Transformers",
+        "ONNX",
+        "safetensors",
+    ],
+    reference="https://huggingface.co/mixedbread-ai/mxbai-rerank-xsmall-v1",
+    similarity_fn_name=None,
+    use_instructions=None,
+    training_datasets=None,
+    adapted_from=None,
+    superseded_by=None,
+    modalities=["text"],
+    model_type=["cross-encoder"],
+    citation="""@online{rerank2024mxbai,
+  title={Boost Your Search With The Crispy Mixedbread Rerank Models},
+  author={Aamir Shakir and Darius Koenig and Julius Lipp and Sean Lee},
+  year={2024},
+  url={https://www.mixedbread.ai/blog/mxbai-rerank-v1},
+}""",
+    contacts=None,
+)
+
+mxbai_rerank_base_v1 = ModelMeta(
+    loader=CrossEncoderWrapper,
+    name="mixedbread-ai/mxbai-rerank-base-v1",
+    revision="800f24c113213a187e65bde9db00c15a2bb12738",
+    release_date="2024-02-29",
+    languages=["eng-Latn"],
+    n_parameters=184422913,
+    memory_usage_mb=352.0,
+    max_tokens=512,
+    embed_dim=None,
+    license="apache-2.0",
+    open_weights=True,
+    public_training_code=None,
+    public_training_data=None,
+    framework=[
+        "PyTorch",
+        "Sentence Transformers",
+        "Transformers",
+        "ONNX",
+        "safetensors",
+    ],
+    reference="https://huggingface.co/mixedbread-ai/mxbai-rerank-base-v1",
+    similarity_fn_name=None,
+    use_instructions=None,
+    training_datasets=None,
+    adapted_from=None,
+    superseded_by=None,
+    modalities=["text"],
+    model_type=["cross-encoder"],
+    citation="""@online{rerank2024mxbai,
+  title={Boost Your Search With The Crispy Mixedbread Rerank Models},
+  author={Aamir Shakir and Darius Koenig and Julius Lipp and Sean Lee},
+  year={2024},
+  url={https://www.mixedbread.ai/blog/mxbai-rerank-v1},
+}""",
+    contacts=None,
+)
+
+mxbai_rerank_large_v1 = ModelMeta(
+    loader=CrossEncoderWrapper,
+    name="mixedbread-ai/mxbai-rerank-large-v1",
+    revision="98f655841d5caf0b16eaff79c2b4ca109d920d17",
+    release_date="2024-02-29",
+    languages=["eng-Latn"],
+    n_parameters=435062785,
+    memory_usage_mb=830.0,
+    max_tokens=512,
+    embed_dim=None,
+    license="apache-2.0",
+    open_weights=True,
+    public_training_code=None,
+    public_training_data=None,
+    framework=[
+        "PyTorch",
+        "Sentence Transformers",
+        "Transformers",
+        "ONNX",
+        "safetensors",
+    ],
+    reference="https://huggingface.co/mixedbread-ai/mxbai-rerank-large-v1",
+    similarity_fn_name=None,
+    use_instructions=None,
+    training_datasets=None,
+    adapted_from=None,
+    superseded_by=None,
+    modalities=["text"],
+    model_type=["cross-encoder"],
+    citation="""@online{rerank2024mxbai,
+  title={Boost Your Search With The Crispy Mixedbread Rerank Models},
+  author={Aamir Shakir and Darius Koenig and Julius Lipp and Sean Lee},
+  year={2024},
+  url={https://www.mixedbread.ai/blog/mxbai-rerank-v1},
+}""",
+    contacts=None,
+)
+
+mxbai_edge_colbert_v0_17m = ModelMeta(
+    loader=MultiVectorModel,
+    name="mixedbread-ai/mxbai-edge-colbert-v0-17m",
+    model_type=["late-interaction"],
+    languages=["eng-Latn"],
+    open_weights=True,
+    revision="23ae07f5bf028bc0d1f80c82e6e2dd2311f13a46",
+    public_training_code=None,
+    public_training_data=None,
+    release_date="2025-10-16",
+    n_parameters=int(17 * 1e6),
+    memory_usage_mb=64,
+    max_tokens=7999,
+    embed_dim=None,
+    license="apache-2.0",
+    similarity_fn_name=ScoringFunction.MAX_SIM,
+    framework=["PyLate", "ColBERT", "Transformers", "safetensors"],
+    reference="https://huggingface.co/mixedbread-ai/mxbai-edge-colbert-v0-17m",
+    use_instructions=False,
+    adapted_from="https://huggingface.co/jhu-clsp/ettin-encoder-17m",
+    superseded_by=None,
+    training_datasets={
+        "CornStack",
+        "MSMARCO",
+        "NQ",
+        "HotpotQA",
+        "AmazonQA",
+        "LoTTE",
+        "MultiLongDocRetrieval",
+        # "FineWeb",
+        # "PubMedQA",
+        # "TriviaQA",
+    },
+    citation="""@misc{takehi2025fantasticsmallretrieverstrain,
+      title={Fantastic (small) Retrievers and How to Train Them: mxbai-edge-colbert-v0 Tech Report},
+      author={Rikiya Takehi and Benjamin Clavié and Sean Lee and Aamir Shakir},
+      year={2025},
+      eprint={2510.14880},
+      archivePrefix={arXiv},
+      primaryClass={cs.IR},
+      url={https://arxiv.org/abs/2510.14880},
+}""",
+    contacts=None,
+)
+
+mxbai_edge_colbert_v0_32m = ModelMeta(
+    loader=MultiVectorModel,
+    name="mixedbread-ai/mxbai-edge-colbert-v0-32m",
+    model_type=["late-interaction"],
+    languages=["eng-Latn"],
+    open_weights=True,
+    revision="2f12870a85dae80680b9babc59992c9a2bc59e4a",
+    public_training_code=None,
+    public_training_data=None,
+    release_date="2025-10-16",
+    n_parameters=int(32 * 1e6),
+    memory_usage_mb=122,
+    max_tokens=511,
+    embed_dim=None,
+    license="apache-2.0",
+    similarity_fn_name=ScoringFunction.MAX_SIM,
+    framework=["PyLate", "ColBERT", "Transformers", "safetensors"],
+    reference="https://huggingface.co/mixedbread-ai/mxbai-edge-colbert-v0-32m",
+    use_instructions=False,
+    adapted_from="https://huggingface.co/jhu-clsp/ettin-encoder-32m",
+    superseded_by=None,
+    training_datasets={
+        "CornStack",
+        "MSMARCO",
+        "NQ",
+        "HotpotQA",
+        "AmazonQA",
+        "LoTTE",
+        "MultiLongDocRetrieval",
+        # "FineWeb",
+        # "PubMedQA",
+        # "TriviaQA",
+    },
+    citation="""@misc{takehi2025fantasticsmallretrieverstrain,
+      title={Fantastic (small) Retrievers and How to Train Them: mxbai-edge-colbert-v0 Tech Report},
+      author={Rikiya Takehi and Benjamin Clavié and Sean Lee and Aamir Shakir},
+      year={2025},
+      eprint={2510.14880},
+      archivePrefix={arXiv},
+      primaryClass={cs.IR},
+      url={https://arxiv.org/abs/2510.14880},
+}""",
+    contacts=None,
+)

mteb/models/model_implementations/mme5_models.py

@@ -25,7 +25,7 @@ mme5_mllama = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data="https://huggingface.co/datasets/intfloat/mmE5-MMEB-hardneg, https://huggingface.co/datasets/intfloat/mmE5-synthetic",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     reference="https://huggingface.co/intfloat/mmE5-mllama-11b-instruct",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=True,

mteb/models/model_implementations/moco_models.py

@@ -117,7 +117,7 @@ mocov3_training_datasets = set(
 )
 
 mocov3_vit_base = ModelMeta(
-    loader=mocov3_loader,  # type: ignore
+    loader=mocov3_loader,
     name="nyu-visionx/moco-v3-vit-b",
     model_type=["dense"],
     languages=["eng-Latn"],
@@ -132,7 +132,7 @@ mocov3_vit_base = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/facebookresearch/moco-v3",
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://github.com/facebookresearch/moco-v3",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -141,7 +141,7 @@ mocov3_vit_base = ModelMeta(
 )
 
 mocov3_vit_large = ModelMeta(
-    loader=mocov3_loader,  # type: ignore
+    loader=mocov3_loader,
     name="nyu-visionx/moco-v3-vit-l",
     model_type=["dense"],
     languages=["eng-Latn"],
@@ -156,7 +156,7 @@ mocov3_vit_large = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/facebookresearch/moco-v3",
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://github.com/facebookresearch/moco-v3",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,

mteb/models/model_implementations/mod_models.py

@@ -181,7 +181,7 @@ MoD_Embedding = ModelMeta(
     license="apache-2.0",
     reference="https://huggingface.co/bflhc/MoD-Embedding",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,

mteb/models/model_implementations/model2vec_models.py

@@ -139,7 +139,7 @@ class Model2VecModel(AbsEncoder):
             **kwargs: Additional arguments to pass to the wrapper.
         """
         requires_package(self, "model2vec", model_name, "pip install 'mteb[model2vec]'")
-        from model2vec import StaticModel  # type: ignore
+        from model2vec import StaticModel
 
         self.model_name = model_name
         self.model = StaticModel.from_pretrained(self.model_name)
@@ -172,7 +172,7 @@ m2v_base_glove_subword = ModelMeta(
     embed_dim=256,
     license="mit",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["NumPy", "Sentence Transformers"],
+    framework=["NumPy", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/minishlab/M2V_base_glove_subword",
     use_instructions=False,
     adapted_from="BAAI/bge-base-en-v1.5",
@@ -198,7 +198,7 @@ m2v_base_glove = ModelMeta(
     embed_dim=256,
     license="mit",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["NumPy", "Sentence Transformers"],
+    framework=["NumPy", "Sentence Transformers", "safetensors"],
     reference="https://huggingface.co/minishlab/M2V_base_glove",
     use_instructions=False,
     adapted_from="BAAI/bge-base-en-v1.5",
@@ -223,7 +223,7 @@ m2v_base_output = ModelMeta(
     embed_dim=256,
     license="mit",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["NumPy", "Sentence Transformers"],
+    framework=["NumPy", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/minishlab/M2V_base_output",
     use_instructions=False,
     adapted_from="BAAI/bge-base-en-v1.5",
@@ -248,7 +248,7 @@ m2v_multilingual_output = ModelMeta(
     embed_dim=256,
     license="mit",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["NumPy", "Sentence Transformers"],
+    framework=["NumPy", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/minishlab/M2V_multilingual_output",
     use_instructions=False,
     adapted_from="sentence-transformers/LaBSE",
@@ -273,7 +273,7 @@ potion_base_2m = ModelMeta(
     embed_dim=64,
     license="mit",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["NumPy", "Sentence Transformers"],
+    framework=["NumPy", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/minishlab/potion-base-2M",
     use_instructions=False,
     adapted_from="BAAI/bge-base-en-v1.5",
@@ -298,7 +298,7 @@ potion_base_4m = ModelMeta(
     embed_dim=128,
     license="mit",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["NumPy", "Sentence Transformers"],
+    framework=["NumPy", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/minishlab/potion-base-4M",
     use_instructions=False,
     adapted_from="BAAI/bge-base-en-v1.5",
@@ -323,7 +323,7 @@ potion_base_8m = ModelMeta(
     embed_dim=256,
     license="mit",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["NumPy", "Sentence Transformers"],
+    framework=["NumPy", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/minishlab/potion-base-8M",
     use_instructions=False,
     adapted_from="BAAI/bge-base-en-v1.5",
@@ -348,7 +348,7 @@ potion_multilingual_128m = ModelMeta(
     embed_dim=256,
     license="mit",
     similarity_fn_name="cosine",
-    framework=["NumPy"],
+    framework=["NumPy", "ONNX", "safetensors", "Sentence Transformers"],
     reference="https://huggingface.co/minishlab/potion-multilingual-128M",
     use_instructions=False,
     adapted_from="BAAI/bge-m3",
@@ -373,7 +373,7 @@ pubmed_bert_100k = ModelMeta(
     embed_dim=64,
     license="apache-2.0",
     similarity_fn_name="cosine",
-    framework=["NumPy"],
+    framework=["NumPy", "Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/NeuML/pubmedbert-base-embeddings-100K",
     use_instructions=False,
     adapted_from="NeuML/pubmedbert-base-embeddings",
@@ -397,7 +397,7 @@ pubmed_bert_500k = ModelMeta(
     embed_dim=64,
     license="apache-2.0",
     similarity_fn_name="cosine",
-    framework=["NumPy"],
+    framework=["NumPy", "Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/NeuML/pubmedbert-base-embeddings-500K",
     use_instructions=False,
     adapted_from="NeuML/pubmedbert-base-embeddings",
@@ -421,7 +421,7 @@ pubmed_bert_1m = ModelMeta(
     embed_dim=64,
     license="apache-2.0",
     similarity_fn_name="cosine",
-    framework=["NumPy"],
+    framework=["NumPy", "Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/NeuML/pubmedbert-base-embeddings-1M",
     use_instructions=False,
     adapted_from="NeuML/pubmedbert-base-embeddings",
@@ -445,7 +445,7 @@ pubmed_bert_2m = ModelMeta(
     embed_dim=64,
     license="apache-2.0",
     similarity_fn_name="cosine",
-    framework=["NumPy"],
+    framework=["NumPy", "Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/NeuML/pubmedbert-base-embeddings-2M",
     use_instructions=False,
     adapted_from="NeuML/pubmedbert-base-embeddings",
@@ -469,7 +469,7 @@ pubmed_bert_8m = ModelMeta(
     embed_dim=256,
     license="apache-2.0",
     similarity_fn_name="cosine",
-    framework=["NumPy"],
+    framework=["NumPy", "Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/NeuML/pubmedbert-base-embeddings-8M",
     use_instructions=False,
     adapted_from="NeuML/pubmedbert-base-embeddings",

mteb/models/model_implementations/moka_models.py

@@ -104,7 +104,7 @@ m3e_base = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/moka-ai/m3e-base",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=False,
     superseded_by=None,
     adapted_from=None,

mteb/models/model_implementations/nbailab.py

@@ -18,7 +18,7 @@ nb_sbert = ModelMeta(
     max_tokens=75,
     reference="https://huggingface.co/NbAiLab/nb-sbert-base",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=False,
     public_training_code=None,
     public_training_data="https://huggingface.co/datasets/NbAiLab/mnli-norwegian",
@@ -40,7 +40,7 @@ nb_bert_large = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/NbAiLab/nb-bert-large",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=False,
     public_training_code=None,
     public_training_data="https://huggingface.co/NbAiLab/nb-bert-large#training-data",
@@ -62,7 +62,7 @@ nb_bert_base = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/NbAiLab/nb-bert-base",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=False,
     public_training_code=None,
     public_training_data="https://huggingface.co/NbAiLab/nb-bert-base#training-data",

mteb/models/model_implementations/no_instruct_sentence_models.py

@@ -30,13 +30,13 @@ class NoInstructModel(AbsEncoder):
         self,
         model_name: str,
         revision: str,
+        device: str | None = None,
         model_prompts: dict[str, str] | None = None,
         **kwargs: Any,
     ):
         from transformers import AutoModel, AutoTokenizer
 
         self.model_name = model_name
-        device = kwargs.pop("device", None)
         self.device = device or ("cuda" if torch.cuda.is_available() else "cpu")
         self.model = AutoModel.from_pretrained(
             model_name, revision=revision, **kwargs
@@ -109,7 +109,7 @@ no_instruct_small_v0 = ModelMeta(
     license="mit",
     reference="https://huggingface.co/avsolatorio/NoInstruct-small-Embedding-v0",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors", "Transformers"],
     use_instructions=False,
     adapted_from=None,
     superseded_by=None,