mteb 2.6.4__py3-none-any.whl → 2.6.6__py3-none-any.whl

mteb/models/model_implementations/bge_models.py

@@ -331,7 +331,13 @@ bge_small_en_v1_5 = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/BAAI/bge-small-en-v1.5",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "ONNX",
+        "safetensors",
+        "Transformers",
+    ],
     use_instructions=True,
     public_training_code=None,
     public_training_data="https://data.baai.ac.cn/details/BAAI-MTP",
@@ -357,7 +363,13 @@ bge_base_en_v1_5 = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/BAAI/bge-base-en-v1.5",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "ONNX",
+        "safetensors",
+        "Transformers",
+    ],
     use_instructions=True,
     public_training_code=None,  # seemingly released (at least for some models, but the link is broken
     public_training_data="https://data.baai.ac.cn/details/BAAI-MTP",
@@ -383,7 +395,13 @@ bge_large_en_v1_5 = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/BAAI/bge-large-en-v1.5",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "ONNX",
+        "safetensors",
+        "Transformers",
+    ],
     use_instructions=True,
     citation=BGE_15_CITATION,
     public_training_code=None,  # seemingly released (at least for some models, but the link is broken
@@ -409,7 +427,7 @@ bge_small_zh = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/BAAI/bge-small-zh",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,
@@ -436,7 +454,7 @@ bge_base_zh = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/BAAI/bge-base-zh",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,
@@ -463,7 +481,7 @@ bge_large_zh = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/BAAI/bge-large-zh",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,
@@ -490,7 +508,7 @@ bge_small_en = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/BAAI/bge-small-en",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     public_training_code=None,
     public_training_data="https://data.baai.ac.cn/details/BAAI-MTP",
@@ -517,7 +535,13 @@ bge_base_en = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/BAAI/bge-base-en",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "Transformers",
+        "ONNX",
+        "safetensors",
+    ],
     use_instructions=True,
     public_training_code=None,  # seemingly released (at least for some models, but the link is broken
     public_training_data="https://data.baai.ac.cn/details/BAAI-MTP",
@@ -544,7 +568,7 @@ bge_large_en = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/BAAI/bge-large-en",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     public_training_code=None,  # seemingly released (at least for some models, but the link is broken
     public_training_data="https://data.baai.ac.cn/details/BAAI-MTP",
@@ -572,7 +596,7 @@ bge_small_zh_v1_5 = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/BAAI/bge-small-zh-v1.5",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,
@@ -598,7 +622,7 @@ bge_base_zh_v1_5 = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/BAAI/bge-base-zh-v1.5",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,
@@ -624,7 +648,7 @@ bge_large_zh_v1_5 = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/BAAI/bge-large-zh-v1.5",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,
@@ -647,13 +671,13 @@ bge_m3 = ModelMeta(
     max_tokens=8194,
     reference="https://huggingface.co/BAAI/bge-m3",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "ONNX"],
     use_instructions=False,
     public_training_code=None,
     public_training_data="https://huggingface.co/datasets/cfli/bge-full-data",
     training_datasets=bge_m3_training_data,
     citation="""@misc{bge-m3,
-      title={BGE M3-Embedding: Multi-Lingual, Multi-Functionality, Multi-Granularity Text Embeddings Through Self-Knowledge Distillation}, 
+      title={BGE M3-Embedding: Multi-Lingual, Multi-Functionality, Multi-Granularity Text Embeddings Through Self-Knowledge Distillation},
       author={Jianlv Chen and Shitao Xiao and Peitian Zhang and Kun Luo and Defu Lian and Zheng Liu},
       year={2024},
       eprint={2402.03216},
@@ -743,7 +767,7 @@ bge_multilingual_gemma2 = ModelMeta(
     max_tokens=8192,  # from old C-MTEB leaderboard
     reference="https://huggingface.co/BAAI/bge-multilingual-gemma2",
     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=None,
@@ -754,7 +778,7 @@ bge_multilingual_gemma2 = ModelMeta(
     | bge_full_data
     | bge_m3_training_data,
     citation="""@misc{bge-m3,
-      title={BGE M3-Embedding: Multi-Lingual, Multi-Functionality, Multi-Granularity Text Embeddings Through Self-Knowledge Distillation}, 
+      title={BGE M3-Embedding: Multi-Lingual, Multi-Functionality, Multi-Granularity Text Embeddings Through Self-Knowledge Distillation},
       author={Jianlv Chen and Shitao Xiao and Peitian Zhang and Kun Luo and Defu Lian and Zheng Liu},
       year={2024},
       eprint={2402.03216},
@@ -764,7 +788,7 @@ bge_multilingual_gemma2 = ModelMeta(
 
 
 @misc{bge_embedding,
-      title={C-Pack: Packaged Resources To Advance General Chinese Embedding}, 
+      title={C-Pack: Packaged Resources To Advance General Chinese Embedding},
       author={Shitao Xiao and Zheng Liu and Peitian Zhang and Niklas Muennighoff},
       year={2023},
       eprint={2309.07597},
@@ -790,7 +814,7 @@ bge_en_icl = ModelMeta(
     max_tokens=32768,
     reference="https://huggingface.co/BAAI/bge-en-icl",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=False,
     public_training_code="https://github.com/FlagOpen/FlagEmbedding",
     public_training_data="https://huggingface.co/datasets/cfli/bge-full-data",
@@ -824,13 +848,13 @@ bge_m3_unsupervised = ModelMeta(
     max_tokens=8192,
     reference="https://huggingface.co/BAAI/bge-m3-unsupervised",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=False,
     public_training_code="https://github.com/FlagOpen/FlagEmbedding",
     public_training_data="https://huggingface.co/datasets/cfli/bge-full-data",
     training_datasets=bge_m3_training_data,
     citation="""@misc{bge-m3,
-      title={BGE M3-Embedding: Multi-Lingual, Multi-Functionality, Multi-Granularity Text Embeddings Through Self-Knowledge Distillation}, 
+      title={BGE M3-Embedding: Multi-Lingual, Multi-Functionality, Multi-Granularity Text Embeddings Through Self-Knowledge Distillation},
       author={Jianlv Chen and Shitao Xiao and Peitian Zhang and Kun Luo and Defu Lian and Zheng Liu},
       year={2024},
       eprint={2402.03216},
@@ -854,7 +878,7 @@ manu__bge_m3_custom_fr = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors"],
     reference="https://huggingface.co/manu/bge-m3-custom-fr",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,

mteb/models/model_implementations/bica_model.py

@@ -15,20 +15,20 @@ bica_base = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/bisectgroup/BiCA-base",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=False,
     public_training_code="https://github.com/NiravBhattLab/BiCA",
     public_training_data="https://huggingface.co/datasets/bisectgroup/hard-negatives-traversal",
     adapted_from="thenlper/gte-base",
     citation="""
 @misc{sinha2025bicaeffectivebiomedicaldense,
-      title={BiCA: Effective Biomedical Dense Retrieval with Citation-Aware Hard Negatives}, 
+      title={BiCA: Effective Biomedical Dense Retrieval with Citation-Aware Hard Negatives},
       author={Aarush Sinha and Pavan Kumar S and Roshan Balaji and Nirav Pravinbhai Bhatt},
       year={2025},
       eprint={2511.08029},
       archivePrefix={arXiv},
       primaryClass={cs.IR},
-      url={https://arxiv.org/abs/2511.08029}, 
+      url={https://arxiv.org/abs/2511.08029},
 }
 """,
     training_datasets=set(),

mteb/models/model_implementations/blip2_models.py

@@ -179,7 +179,7 @@ blip2_opt_2_7b = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/salesforce/LAVIS/tree/main/projects/blip2",
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://huggingface.co/Salesforce/blip2-opt-2.7b",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -203,7 +203,7 @@ blip2_opt_6_7b_coco = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/salesforce/LAVIS/tree/main/projects/blip2",
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://huggingface.co/Salesforce/blip2-opt-6.7b-coco",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,

mteb/models/model_implementations/blip_models.py

@@ -143,7 +143,7 @@ blip_image_captioning_large = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/salesforce/BLIP",
     public_training_data="https://github.com/salesforce/BLIP",
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://huggingface.co/Salesforce/blip-image-captioning-large",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -171,7 +171,7 @@ blip_image_captioning_base = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/salesforce/BLIP",
     public_training_data="https://github.com/salesforce/BLIP",
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/Salesforce/blip-image-captioning-base",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -200,7 +200,7 @@ blip_vqa_base = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/salesforce/BLIP",
     public_training_data="https://github.com/salesforce/BLIP",
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://huggingface.co/Salesforce/blip-vqa-base",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -227,7 +227,7 @@ blip_vqa_capfilt_large = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/salesforce/BLIP",
     public_training_data="https://github.com/salesforce/BLIP",
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/Salesforce/blip-vqa-capfilt-large",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -254,7 +254,7 @@ blip_itm_base_coco = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/salesforce/BLIP",
     public_training_data="https://github.com/salesforce/BLIP",
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/Salesforce/blip-itm-base-coco",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -281,7 +281,7 @@ blip_itm_large_coco = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/salesforce/BLIP",
     public_training_data="https://github.com/salesforce/BLIP",
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/Salesforce/blip-itm-large-coco",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -309,7 +309,7 @@ blip_itm_base_flickr = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/salesforce/BLIP",
     public_training_data="https://github.com/salesforce/BLIP",
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/Salesforce/blip-itm-base-flickr",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -337,7 +337,7 @@ blip_itm_large_flickr = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/salesforce/BLIP",
     public_training_data="https://github.com/salesforce/BLIP",
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/Salesforce/blip-itm-large-flickr",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,

mteb/models/model_implementations/bmretriever_models.py

@@ -104,7 +104,7 @@ BMRetriever_410M = ModelMeta(
     license="mit",
     reference="https://huggingface.co/BMRetriever/BMRetriever-410M",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,
@@ -134,7 +134,7 @@ BMRetriever_1B = ModelMeta(
     license="mit",
     reference="https://huggingface.co/BMRetriever/BMRetriever-1B",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,
@@ -164,7 +164,7 @@ BMRetriever_2B = ModelMeta(
     license="mit",
     reference="https://huggingface.co/BMRetriever/BMRetriever-2B",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,
@@ -194,7 +194,7 @@ BMRetriever_7B = ModelMeta(
     license="mit",
     reference="https://huggingface.co/BMRetriever/BMRetriever-7B",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,

mteb/models/model_implementations/cadet_models.py

@@ -47,7 +47,7 @@ cadet_embed = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/manveertamber/cadet-embed-base-v1",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=True,
     public_training_code="https://github.com/manveertamber/cadet-dense-retrieval",
     # we provide the code to generate the training data

mteb/models/model_implementations/cde_models.py

@@ -227,7 +227,7 @@ cde_small_v1 = ModelMeta(
     embed_dim=768,
     license="mit",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers"],
+    framework=["Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/jxm/cde-small-v1",
     use_instructions=True,
     adapted_from="nomic-ai/nomic-bert-2048",
@@ -256,7 +256,7 @@ cde_small_v2 = ModelMeta(
     embed_dim=768,
     license="mit",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers"],
+    framework=["Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/jxm/cde-small-v1",
     use_instructions=True,
     adapted_from="answerdotai/ModernBERT-base",

mteb/models/model_implementations/clip_models.py

@@ -130,7 +130,7 @@ clip_vit_large_patch14 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://huggingface.co/openai/clip-vit-large-patch14",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -154,7 +154,7 @@ clip_vit_base_patch32 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/openai/clip-vit-base-patch32",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -178,7 +178,7 @@ clip_vit_base_patch16 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/openai/clip-vit-base-patch16",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,

mteb/models/model_implementations/clips_models.py

@@ -36,7 +36,7 @@ e5_nl_small = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/clips/e5-small-trm-nl",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code="https://github.com/ELotfi/e5-nl",
     public_training_data="https://huggingface.co/collections/clips/beir-nl",
@@ -63,7 +63,7 @@ e5_nl_base = ModelMeta(
     max_tokens=514,
     reference="https://huggingface.co/clips/e5-base-trm-nl",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code="https://github.com/ELotfi/e5-nl",
     public_training_data="https://huggingface.co/collections/clips/beir-nl",
@@ -90,7 +90,7 @@ e5_nl_large = ModelMeta(
     max_tokens=514,
     reference="https://huggingface.co/clips/e5-large-trm-nl",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code="https://github.com/ELotfi/e5-nl",
     public_training_data="https://huggingface.co/collections/clips/beir-nl",

mteb/models/model_implementations/codefuse_models.py

@@ -242,7 +242,7 @@ F2LLM_0B6 = ModelMeta(
     max_tokens=8192,
     reference="https://huggingface.co/codefuse-ai/F2LLM-0.6B",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code="https://github.com/codefuse-ai/F2LLM",
     public_training_data="https://huggingface.co/datasets/codefuse-ai/F2LLM",
@@ -272,7 +272,7 @@ F2LLM_1B7 = ModelMeta(
     max_tokens=8192,
     reference="https://huggingface.co/codefuse-ai/F2LLM-1.7B",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code="https://github.com/codefuse-ai/F2LLM",
     public_training_data="https://huggingface.co/datasets/codefuse-ai/F2LLM",
@@ -302,7 +302,7 @@ F2LLM_4B = ModelMeta(
     max_tokens=8192,
     reference="https://huggingface.co/codefuse-ai/F2LLM-4B",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code="https://github.com/codefuse-ai/F2LLM",
     public_training_data="https://huggingface.co/datasets/codefuse-ai/F2LLM",
@@ -325,7 +325,7 @@ C2LLM_0B5 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "Transformers", "safetensors"],
     reference="https://huggingface.co/codefuse-ai/C2LLM-0.5B",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=True,
@@ -353,7 +353,7 @@ C2LLM_7B = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "Transformers", "safetensors"],
     reference="https://huggingface.co/codefuse-ai/C2LLM-7B",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=True,

mteb/models/model_implementations/codesage_models.py

@@ -35,7 +35,7 @@ codesage_large = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/codesage/codesage-large-v2",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -62,7 +62,7 @@ codesage_base = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/codesage/codesage-base-v2",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,
@@ -89,7 +89,7 @@ codesage_small = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/codesage/codesage-small-v2",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=False,

mteb/models/model_implementations/cohere_models.py

@@ -392,7 +392,7 @@ cohere_mult_3 = ModelMeta(
     reference="https://cohere.com/blog/introducing-embed-v3",
     license=None,
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["API"],
+    framework=["API", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,  # assumed
@@ -417,7 +417,7 @@ cohere_eng_3 = ModelMeta(
     embed_dim=1024,
     license=None,
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["API"],
+    framework=["API", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,  # assumed
@@ -442,7 +442,7 @@ cohere_mult_light_3 = ModelMeta(
     embed_dim=384,
     license=None,
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["API"],
+    framework=["API", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,  # assumed
@@ -467,7 +467,7 @@ cohere_eng_light_3 = ModelMeta(
     embed_dim=384,
     license=None,
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["API"],
+    framework=["API", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,  # assumed

mteb/models/model_implementations/colpali_models.py

@@ -226,7 +226,7 @@ colpali_v1_1 = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/illuin-tech/colpali",
     public_training_data="https://huggingface.co/datasets/vidore/colpali_train_set",
-    framework=["ColPali"],
+    framework=["ColPali", "safetensors"],
     reference="https://huggingface.co/vidore/colpali-v1.1",
     similarity_fn_name=ScoringFunction.MAX_SIM,
     use_instructions=True,
@@ -253,7 +253,7 @@ colpali_v1_2 = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/illuin-tech/colpali",
     public_training_data="https://huggingface.co/datasets/vidore/colpali_train_set",
-    framework=["ColPali"],
+    framework=["ColPali", "safetensors"],
     reference="https://huggingface.co/vidore/colpali-v1.2",
     similarity_fn_name=ScoringFunction.MAX_SIM,
     use_instructions=True,
@@ -280,7 +280,7 @@ colpali_v1_3 = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/illuin-tech/colpali",
     public_training_data="https://huggingface.co/datasets/vidore/colpali_train_set",
-    framework=["ColPali"],
+    framework=["ColPali", "safetensors"],
     reference="https://huggingface.co/vidore/colpali-v1.3",
     similarity_fn_name=ScoringFunction.MAX_SIM,
     use_instructions=True,

mteb/models/model_implementations/colqwen_models.py

@@ -226,7 +226,7 @@ colqwen2 = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/illuin-tech/colpali",
     public_training_data="https://huggingface.co/datasets/vidore/colpali_train_set",
-    framework=["ColPali"],
+    framework=["ColPali", "safetensors"],
     reference="https://huggingface.co/vidore/colqwen2-v1.0",
     similarity_fn_name="MaxSim",
     use_instructions=True,
@@ -253,7 +253,7 @@ colqwen2_5 = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/illuin-tech/colpali",
     public_training_data="https://huggingface.co/datasets/vidore/colpali_train_set",
-    framework=["ColPali"],
+    framework=["ColPali", "safetensors"],
     reference="https://huggingface.co/vidore/colqwen2.5-v0.2",
     similarity_fn_name="MaxSim",
     use_instructions=True,
@@ -297,7 +297,7 @@ colqwen3_8b = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/illuin-tech/colpali",
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://huggingface.co/TomoroAI/tomoro-colqwen3-embed-8b",
     similarity_fn_name=ScoringFunction.MAX_SIM,
     use_instructions=True,
@@ -321,7 +321,7 @@ colqwen3_4b = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/illuin-tech/colpali",
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://huggingface.co/TomoroAI/tomoro-colqwen3-embed-4b",
     similarity_fn_name=ScoringFunction.MAX_SIM,
     use_instructions=True,
@@ -348,7 +348,7 @@ colnomic_7b = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/nomic-ai/colpali",
     public_training_data="https://huggingface.co/datasets/vidore/colpali_train_set",
-    framework=["ColPali"],
+    framework=["ColPali", "safetensors"],
     reference="https://huggingface.co/nomic-ai/colnomic-embed-multimodal-7b",
     similarity_fn_name="MaxSim",
     use_instructions=True,
@@ -393,7 +393,7 @@ colnomic_3b = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/nomic-ai/colpali",
     public_training_data="https://huggingface.co/datasets/vidore/colpali_train_set",
-    framework=["ColPali"],
+    framework=["ColPali", "safetensors"],
     reference="https://huggingface.co/nomic-ai/colnomic-embed-multimodal-3b",
     similarity_fn_name="MaxSim",
     use_instructions=True,
@@ -458,7 +458,7 @@ evoqwen25_vl_retriever_3b_v1 = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/illuin-tech/colpali",
     public_training_data="https://huggingface.co/datasets/vidore/colpali_train_set",
-    framework=["ColPali"],
+    framework=["ColPali", "safetensors"],
     reference="https://huggingface.co/ApsaraStackMaaS/EvoQwen2.5-VL-Retriever-3B-v1",
     similarity_fn_name="MaxSim",
     use_instructions=True,
@@ -484,7 +484,7 @@ evoqwen25_vl_retriever_7b_v1 = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/illuin-tech/colpali",
     public_training_data="https://huggingface.co/datasets/vidore/colpali_train_set",
-    framework=["ColPali"],
+    framework=["ColPali", "safetensors"],
     reference="https://huggingface.co/ApsaraStackMaaS/EvoQwen2.5-VL-Retriever-7B-v1",
     similarity_fn_name="MaxSim",
     use_instructions=True,