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

mteb/models/model_implementations/euler_models.py

@@ -16,7 +16,7 @@ Euler_Legal_Embedding_V1 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/Mira190/Euler-Legal-Embedding-V1",
     similarity_fn_name="cosine",
     use_instructions=False,
@@ -24,7 +24,7 @@ Euler_Legal_Embedding_V1 = ModelMeta(
     adapted_from="Qwen/Qwen3-Embedding-8B",
     superseded_by=None,
     citation="""@misc{euler2025legal,
-      title={Euler-Legal-Embedding: Advanced Legal Representation Learning}, 
+      title={Euler-Legal-Embedding: Advanced Legal Representation Learning},
       author={LawRank Team},
       year={2025},
       publisher={Hugging Face}

mteb/models/model_implementations/fa_models.py

@@ -18,7 +18,7 @@ parsbert = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/HooshvareLab/bert-base-parsbert-uncased",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers"],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,
@@ -54,7 +54,7 @@ bert_zwnj = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/m3hrdadfi/bert-zwnj-wnli-mean-tokens",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers"],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,
@@ -80,7 +80,7 @@ roberta_zwnj = ModelMeta(
     max_tokens=514,
     reference="https://huggingface.co/m3hrdadfi/roberta-zwnj-wnli-mean-tokens",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers"],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,
@@ -105,7 +105,7 @@ sentence_transformer_parsbert = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/myrkur/sentence-transformer-parsbert-fa",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,
@@ -129,7 +129,7 @@ tooka_bert_base = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/PartAI/TookaBERT-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=None,
@@ -156,7 +156,7 @@ tooka_sbert = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/PartAI/Tooka-SBERT",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,
@@ -187,7 +187,7 @@ fa_bert = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/sbunlp/fabert",
     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=None,
@@ -235,7 +235,7 @@ tooka_sbert_v2_small = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/PartAI/Tooka-SBERT-V2-Small",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,
@@ -266,7 +266,7 @@ tooka_sbert_v2_large = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/PartAI/Tooka-SBERT-V2-Large",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,

mteb/models/model_implementations/facebookai.py

@@ -119,7 +119,13 @@ xlmr_base = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/FacebookAI/xlm-roberta-base",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "Transformers",
+        "ONNX",
+        "safetensors",
+    ],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,
@@ -163,7 +169,13 @@ xlmr_large = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/FacebookAI/xlm-roberta-large",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "Transformers",
+        "ONNX",
+        "safetensors",
+    ],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,

mteb/models/model_implementations/geogpt_models.py

@@ -26,7 +26,7 @@ geoembedding = ModelMeta(
     max_tokens=32768,
     reference="https://huggingface.co/GeoGPT-Research-Project/GeoEmbedding",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,

mteb/models/model_implementations/gme_v_models.py

@@ -360,7 +360,7 @@ gme_qwen2vl_2b = ModelMeta(
     max_tokens=32768,
     reference="https://huggingface.co/Alibaba-NLP/gme-Qwen2-VL-2B-Instruct",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,
@@ -384,7 +384,7 @@ gme_qwen2vl_7b = ModelMeta(
     max_tokens=32768,
     reference="https://huggingface.co/Alibaba-NLP/gme-Qwen2-VL-7B-Instruct",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,

mteb/models/model_implementations/google_models.py

@@ -270,7 +270,7 @@ embedding_gemma_300m = ModelMeta(
     max_tokens=2048,
     license="gemma",
     reference="https://ai.google.dev/gemma/docs/embeddinggemma/model_card",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,

mteb/models/model_implementations/granite_vision_embedding_models.py

@@ -179,7 +179,7 @@ granite_vision_embedding = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://huggingface.co/ibm-granite/granite-vision-3.3-2b-embedding",
     similarity_fn_name="MaxSim",
     use_instructions=True,

mteb/models/model_implementations/gritlm_models.py

@@ -50,7 +50,7 @@ gritlm7b = ModelMeta(
     max_tokens=32768,
     reference="https://huggingface.co/GritLM/GritLM-7B",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["GritLM", "PyTorch"],
+    framework=["GritLM", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     training_datasets=GRIT_LM_TRAINING_DATA,
     # section 3.1 "We finetune our final models from Mistral 7B [68] and Mixtral 8x7B [69] using adaptations of E5 [160] and the Tülu 2 data
@@ -79,7 +79,7 @@ gritlm8x7b = ModelMeta(
     max_tokens=32768,
     reference="https://huggingface.co/GritLM/GritLM-8x7B",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["GritLM", "PyTorch"],
+    framework=["GritLM", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     training_datasets=GRIT_LM_TRAINING_DATA,
     citation=GRITLM_CITATION,

mteb/models/model_implementations/gte_models.py

@@ -53,7 +53,7 @@ gte_qwen2_7b_instruct = ModelMeta(
     license="apache-2.0",
     reference="https://huggingface.co/Alibaba-NLP/gte-Qwen2-7B-instruct",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     citation=GTE_CITATION,
     public_training_code=None,
@@ -86,7 +86,7 @@ gte_qwen1_5_7b_instruct = ModelMeta(
     max_tokens=32_768,
     reference="https://huggingface.co/Alibaba-NLP/gte-Qwen1.5-7B-instruct",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,
@@ -123,7 +123,7 @@ gte_qwen2_1_5b_instruct = ModelMeta(
     max_tokens=32_768,
     reference="https://huggingface.co/Alibaba-NLP/gte-Qwen2-1.5B-instruct",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,
@@ -151,7 +151,7 @@ gte_small_zh = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/thenlper/gte-small-zh",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,
@@ -179,7 +179,7 @@ gte_base_zh = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/thenlper/gte-base-zh",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,
@@ -207,7 +207,7 @@ gte_large_zh = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/thenlper/gte-large-zh",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,
@@ -336,7 +336,7 @@ gte_multilingual_base = ModelMeta(
     max_tokens=8192,
     reference="https://huggingface.co/Alibaba-NLP/gte-multilingual-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=None,  # couldn't find
@@ -365,7 +365,13 @@ gte_modernbert_base = ModelMeta(
     max_tokens=8192,
     reference="https://huggingface.co/Alibaba-NLP/gte-modernbert-base",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "Transformers",
+        "ONNX",
+        "safetensors",
+    ],
     use_instructions=False,
     public_training_code=None,  # couldn't find
     public_training_data=None,
@@ -402,7 +408,13 @@ gte_base_en_v15 = ModelMeta(
     max_tokens=8192,
     reference="https://huggingface.co/Alibaba-NLP/gte-base-en-v1.5",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "Transformers",
+        "ONNX",
+        "safetensors",
+    ],
     use_instructions=False,
     superseded_by=None,
     adapted_from=None,
@@ -410,21 +422,21 @@ gte_base_en_v15 = ModelMeta(
     public_training_data=None,
     training_datasets=None,
     citation="""@misc{zhang2024mgte,
-  title={mGTE: Generalized Long-Context Text Representation and Reranking Models for Multilingual Text Retrieval}, 
+  title={mGTE: Generalized Long-Context Text Representation and Reranking Models for Multilingual Text Retrieval},
   author={Xin Zhang and Yanzhao Zhang and Dingkun Long and Wen Xie and Ziqi Dai and Jialong Tang and Huan Lin and Baosong Yang and Pengjun Xie and Fei Huang and Meishan Zhang and Wenjie Li and Min Zhang},
   year={2024},
   eprint={2407.19669},
   archivePrefix={arXiv},
   primaryClass={cs.CL},
-  url={https://arxiv.org/abs/2407.19669}, 
+  url={https://arxiv.org/abs/2407.19669},
 }
 @misc{li2023gte,
-  title={Towards General Text Embeddings with Multi-stage Contrastive Learning}, 
+  title={Towards General Text Embeddings with Multi-stage Contrastive Learning},
   author={Zehan Li and Xin Zhang and Yanzhao Zhang and Dingkun Long and Pengjun Xie and Meishan Zhang},
   year={2023},
   eprint={2308.03281},
   archivePrefix={arXiv},
   primaryClass={cs.CL},
-  url={https://arxiv.org/abs/2308.03281}, 
+  url={https://arxiv.org/abs/2308.03281},
 }""",
 )

mteb/models/model_implementations/hinvec_models.py

@@ -49,7 +49,7 @@ Hinvec_bidir = ModelMeta(
     max_tokens=2048,
     reference="https://huggingface.co/Sailesh97/Hinvec",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     training_datasets=hinvec_training_datasets,
     public_training_code=None,

mteb/models/model_implementations/ibm_granite_models.py

@@ -106,7 +106,13 @@ granite_107m_multilingual = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/ibm-granite/granite-embedding-107m-multilingual",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "Transformers",
+        "ONNX",
+        "safetensors",
+    ],
     adapted_from=None,
     superseded_by=None,
     public_training_code=None,
@@ -131,7 +137,13 @@ granite_278m_multilingual = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/ibm-granite/granite-embedding-278m-multilingual",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "Transformers",
+        "ONNX",
+        "safetensors",
+    ],
     adapted_from=None,
     superseded_by=None,
     public_training_code=None,
@@ -156,7 +168,13 @@ granite_30m_english = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/ibm-granite/granite-embedding-30m-english",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "ONNX",
+        "safetensors",
+        "Transformers",
+    ],
     adapted_from=None,
     superseded_by=None,
     public_training_code=None,
@@ -181,7 +199,13 @@ granite_125m_english = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/ibm-granite/granite-embedding-125m-english",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "ONNX",
+        "safetensors",
+        "Transformers",
+    ],
     adapted_from=None,
     superseded_by=None,
     public_training_code=None,
@@ -207,7 +231,7 @@ granite_english_r2 = ModelMeta(
     max_tokens=8192,
     reference="https://huggingface.co/ibm-granite/granite-embedding-english-r2",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     adapted_from=None,
     superseded_by=None,
     public_training_code=None,
@@ -232,7 +256,7 @@ granite_small_english_r2 = ModelMeta(
     max_tokens=8192,
     reference="https://huggingface.co/ibm-granite/granite-embedding-small-english-r2",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     adapted_from=None,
     superseded_by=None,
     public_training_code=None,

mteb/models/model_implementations/inf_models.py

@@ -62,7 +62,7 @@ inf_retriever_v1 = ModelMeta(
     max_tokens=32768,
     reference="https://huggingface.co/infly/inf-retriever-v1",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     adapted_from="Alibaba-NLP/gte-Qwen2-7B-instruct",
     public_training_code=None,
@@ -89,7 +89,7 @@ inf_retriever_v1_1_5b = ModelMeta(
     max_tokens=32768,
     reference="https://huggingface.co/infly/inf-retriever-v1-1.5b",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     adapted_from="Alibaba-NLP/gte-Qwen2-1.5B-instruct",
     public_training_code=None,

mteb/models/model_implementations/jasper_models.py

@@ -298,7 +298,7 @@ jasper_en_v1 = ModelMeta(
     license="apache-2.0",
     reference="https://huggingface.co/infgrad/jasper_en_vision_language_v1",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=True,
     adapted_from=None,
     superseded_by=None,
@@ -345,7 +345,7 @@ Jasper_Token_Compression_600M = ModelMeta(
     max_tokens=32768,
     reference="https://huggingface.co/infgrad/Jasper-Token-Compression-600M",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=True,
     public_training_code="https://github.com/DunZhang/Jasper-Token-Compression-Training",
     # public_training_data: unsupervised data for distillation

mteb/models/model_implementations/jina_clip.py

@@ -136,7 +136,7 @@ jina_clip_v1 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/jinaai/jina-clip-v1",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=True,

mteb/models/model_implementations/jina_models.py

@@ -733,7 +733,7 @@ jina_reranker_v3 = ModelMeta(
     embed_dim=None,
     license="cc-by-nc-4.0",
     similarity_fn_name=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     use_instructions=None,
     reference="https://huggingface.co/jinaai/jina-reranker-v3",
     public_training_code=None,
@@ -776,7 +776,7 @@ jina_embeddings_v4 = ModelMeta(
     embed_dim=2048,
     license="cc-by-nc-4.0",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=True,
     reference="https://huggingface.co/jinaai/jina-embeddings-v4",
     public_training_code=None,
@@ -824,7 +824,13 @@ jina_embeddings_v3 = ModelMeta(
     embed_dim=1024,
     license="cc-by-nc-4.0",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "Transformers",
+        "ONNX",
+        "safetensors",
+    ],
     use_instructions=True,
     reference="https://huggingface.co/jinaai/jina-embeddings-v3",
     public_training_code=None,
@@ -879,7 +885,7 @@ jina_embeddings_v2_base_en = ModelMeta(
     max_tokens=8192,
     reference="https://huggingface.co/jinaai/jina-embeddings-v2-base-en",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "ONNX", "safetensors"],
     use_instructions=False,
     superseded_by=None,
     adapted_from="jina-bert-base-en-v1",  # pretrained on C4 with Alibi to support longer context.
@@ -943,7 +949,7 @@ jina_embeddings_v2_small_en = ModelMeta(
     max_tokens=8192,
     reference="https://huggingface.co/jinaai/jina-embeddings-v2-small-en",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "ONNX", "safetensors"],
     use_instructions=False,
     superseded_by=None,
     adapted_from="jina-bert-smalll-en-v1",  # pretrained on C4 with Alibi to support longer context

mteb/models/model_implementations/kblab.py

@@ -16,16 +16,22 @@ sbert_swedish = ModelMeta(
     max_tokens=384,
     reference="https://huggingface.co/KBLab/sentence-bert-swedish-cased",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=[
+        "Sentence Transformers",
+        "PyTorch",
+        "safetensors",
+        "GGUF",
+        "Transformers",
+    ],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,
     training_datasets=None,
     adapted_from="sentence-transformers/all-mpnet-base-v2",
-    citation="""@misc{rekathati2021introducing,  
-  author = {Rekathati, Faton},  
-  title = {The KBLab Blog: Introducing a Swedish Sentence Transformer},  
-  url = {https://kb-labb.github.io/posts/2021-08-23-a-swedish-sentence-transformer/},  
-  year = {2021}  
+    citation="""@misc{rekathati2021introducing,
+  author = {Rekathati, Faton},
+  title = {The KBLab Blog: Introducing a Swedish Sentence Transformer},
+  url = {https://kb-labb.github.io/posts/2021-08-23-a-swedish-sentence-transformer/},
+  year = {2021}
 }""",
 )

mteb/models/model_implementations/kennethenevoldsen_models.py

@@ -18,7 +18,7 @@ dfm_enc_large = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/KennethEnevoldsen/dfm-sentence-encoder-large",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers", "safetensors"],
     use_instructions=False,
     superseded_by=None,
     adapted_from="chcaa/dfm-encoder-large-v1",
@@ -53,7 +53,7 @@ dfm_enc_med = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/KennethEnevoldsen/dfm-sentence-encoder-medium",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "Transformers"],
     use_instructions=False,
     superseded_by=None,
     adapted_from=None,

mteb/models/model_implementations/kfst.py

@@ -16,7 +16,7 @@ xlmr_scandi = ModelMeta(
     max_tokens=512,
     reference="https://huggingface.co/KFST/XLMRoberta-en-da-sv-nb",
     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,

mteb/models/model_implementations/kowshik24_models.py

@@ -15,7 +15,7 @@ kowshik24_bangla_embedding_model = ModelMeta(
     max_tokens=128,
     reference="https://huggingface.co/Kowshik24/bangla-sentence-transformer-ft-matryoshka-paraphrase-multilingual-mpnet-base-v2",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors"],
     use_instructions=False,
     public_training_code="https://github.com/kowshik24/Bangla-Embedding",
     public_training_data="https://huggingface.co/datasets/sartajekram/BanglaRQA",

mteb/models/model_implementations/lgai_embedding_models.py

@@ -58,7 +58,7 @@ lgai_embedding_en = ModelMeta(
     max_tokens=32768,
     reference="https://huggingface.co/annamodels/LGAI-Embedding-Preview",
     similarity_fn_name="cosine",
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,

mteb/models/model_implementations/linq_models.py

@@ -44,7 +44,7 @@ Linq_Embed_Mistral = ModelMeta(
     max_tokens=32768,
     reference="https://huggingface.co/Linq-AI-Research/Linq-Embed-Mistral",
     similarity_fn_name=ScoringFunction.COSINE,
-    framework=["Sentence Transformers", "PyTorch"],
+    framework=["Sentence Transformers", "PyTorch", "safetensors", "Transformers"],
     use_instructions=True,
     public_training_code=None,
     public_training_data=None,

mteb/models/model_implementations/listconranker.py

@@ -125,7 +125,7 @@ listconranker = ModelMeta(
     license="mit",
     max_tokens=512,
     reference="https://huggingface.co/ByteDance/ListConRanker",
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors", "Transformers"],
     use_instructions=False,
     public_training_code=None,
     public_training_data=None,

mteb/models/model_implementations/llm2clip_models.py

@@ -196,7 +196,7 @@ llm2clip_openai_l_14_336 = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/microsoft/LLM2CLIP",
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "safetensors"],
     reference="https://huggingface.co/microsoft/LLM2CLIP-Openai-L-14-336",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=True,
@@ -221,7 +221,7 @@ llm2clip_openai_l_14_224 = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/microsoft/LLM2CLIP",
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "safetensors"],
     reference="https://huggingface.co/microsoft/LLM2CLIP-Openai-L-14-224",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=True,
@@ -245,7 +245,7 @@ llm2clip_openai_b_16 = ModelMeta(
     open_weights=True,
     public_training_code="https://github.com/microsoft/LLM2CLIP",
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "safetensors"],
     reference="https://huggingface.co/microsoft/LLM2CLIP-Openai-B-16",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=True,