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

mteb/models/model_implementations/misc_models.py

@@ -25,7 +25,7 @@ Haon_Chen__speed_embedding_7b_instruct = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://huggingface.co/Haon-Chen/speed-embedding-7b-instruct",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -54,7 +54,7 @@ Gameselo__STS_multilingual_mpnet_base_v2 = 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/Gameselo/STS-multilingual-mpnet-base-v2",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -155,7 +155,7 @@ Hum_Works__lodestone_base_4096_v1 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers"],
     reference="https://huggingface.co/Hum-Works/lodestone-base-4096-v1",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -222,7 +222,7 @@ Jaume__gemma_2b_embeddings = 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/Jaume/gemma-2b-embeddings",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -257,7 +257,7 @@ Lajavaness__bilingual_embedding_base = 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/Lajavaness/bilingual-embedding-base",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -306,7 +306,7 @@ Lajavaness__bilingual_embedding_large = 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/Lajavaness/bilingual-embedding-large",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -355,7 +355,7 @@ Lajavaness__bilingual_embedding_small = 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/Lajavaness/bilingual-embedding-small",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -401,7 +401,7 @@ Mihaiii__Bulbasaur = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/Mihaiii/Bulbasaur",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -425,7 +425,7 @@ Mihaiii__Ivysaur = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/Mihaiii/Ivysaur",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -449,7 +449,7 @@ Mihaiii__Squirtle = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/Mihaiii/Squirtle",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -473,7 +473,7 @@ Mihaiii__Venusaur = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/Mihaiii/Venusaur",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -497,7 +497,7 @@ Mihaiii__Wartortle = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/Mihaiii/Wartortle",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -521,7 +521,7 @@ Mihaiii__gte_micro = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/Mihaiii/gte-micro",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -544,7 +544,7 @@ Mihaiii__gte_micro_v4 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/Mihaiii/gte-micro-v4",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -567,7 +567,7 @@ OrdalieTech__Solon_embeddings_large_0_1 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://huggingface.co/OrdalieTech/Solon-embeddings-large-0.1",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -590,7 +590,7 @@ Omartificial_Intelligence_Space__Arabert_all_nli_triplet_Matryoshka = 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/Omartificial-Intelligence-Space/Arabert-all-nli-triplet-Matryoshka",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -622,7 +622,7 @@ Omartificial_Intelligence_Space__Arabic_MiniLM_L12_v2_all_nli_triplet = ModelMet
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors"],
     reference="https://huggingface.co/Omartificial-Intelligence-Space/Arabic-MiniLM-L12-v2-all-nli-triplet",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -647,7 +647,7 @@ Omartificial_Intelligence_Space__Arabic_all_nli_triplet_Matryoshka = 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/Omartificial-Intelligence-Space/Arabic-all-nli-triplet-Matryoshka",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -681,7 +681,7 @@ Omartificial_Intelligence_Space__Arabic_labse_Matryoshka = 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/Omartificial-Intelligence-Space/Arabic-labse-Matryoshka",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -715,7 +715,7 @@ Omartificial_Intelligence_Space__Arabic_mpnet_base_all_nli_triplet = 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/Omartificial-Intelligence-Space/Arabic-mpnet-base-all-nli-triplet",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -749,7 +749,7 @@ Omartificial_Intelligence_Space__Marbert_all_nli_triplet_Matryoshka = 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/Omartificial-Intelligence-Space/Marbert-all-nli-triplet-Matryoshka",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -781,7 +781,7 @@ consciousai__cai_lunaris_text_embeddings = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "Transformers"],
     reference="https://huggingface.co/consciousAI/cai-lunaris-text-embeddings",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -804,7 +804,7 @@ consciousai__cai_stellaris_text_embeddings = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers"],
     reference="https://huggingface.co/consciousAI/cai-stellaris-text-embeddings",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -836,7 +836,7 @@ manu__sentence_croissant_alpha_v0_2 = 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/sentence_croissant_alpha_v0.2",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -859,7 +859,7 @@ manu__sentence_croissant_alpha_v0_3 = 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/sentence_croissant_alpha_v0.3",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -882,7 +882,7 @@ manu__sentence_croissant_alpha_v0_4 = 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/sentence_croissant_alpha_v0.4",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -906,7 +906,7 @@ thenlper__gte_base = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/thenlper/gte-base",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -935,7 +935,7 @@ thenlper__gte_large = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/thenlper/gte-large",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -964,7 +964,7 @@ thenlper__gte_small = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/thenlper/gte-small",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1039,7 +1039,7 @@ sdadas__mmlw_e5_base = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/sdadas/mmlw-e5-base",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1047,7 +1047,7 @@ sdadas__mmlw_e5_base = ModelMeta(
     adapted_from="intfloat/multilingual-e5-base",
     superseded_by=None,
     citation="""@article{dadas2024pirb,
-  title={{PIRB}: A Comprehensive Benchmark of Polish Dense and Hybrid Text Retrieval Methods}, 
+  title={{PIRB}: A Comprehensive Benchmark of Polish Dense and Hybrid Text Retrieval Methods},
   author={Sławomir Dadas and Michał Perełkiewicz and Rafał Poświata},
   year={2024},
   eprint={2402.13350},
@@ -1070,7 +1070,7 @@ dwzhu__e5_base_4k = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/dwzhu/e5-base-4k",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1099,7 +1099,7 @@ sdadas__mmlw_e5_large = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/sdadas/mmlw-e5-large",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1107,7 +1107,7 @@ sdadas__mmlw_e5_large = ModelMeta(
     adapted_from="intfloat/multilingual-e5-large",
     superseded_by=None,
     citation="""@article{dadas2024pirb,
-  title={{PIRB}: A Comprehensive Benchmark of Polish Dense and Hybrid Text Retrieval Methods}, 
+  title={{PIRB}: A Comprehensive Benchmark of Polish Dense and Hybrid Text Retrieval Methods},
   author={Sławomir Dadas and Michał Perełkiewicz and Rafał Poświata},
   year={2024},
   eprint={2402.13350},
@@ -1130,7 +1130,7 @@ sdadas__mmlw_e5_small = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/sdadas/mmlw-e5-small",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1138,7 +1138,7 @@ sdadas__mmlw_e5_small = ModelMeta(
     adapted_from="intfloat/multilingual-e5-small",
     superseded_by=None,
     citation="""@article{dadas2024pirb,
-  title={{PIRB}: A Comprehensive Benchmark of Polish Dense and Hybrid Text Retrieval Methods}, 
+  title={{PIRB}: A Comprehensive Benchmark of Polish Dense and Hybrid Text Retrieval Methods},
   author={Sławomir Dadas and Michał Perełkiewicz and Rafał Poświata},
   year={2024},
   eprint={2402.13350},
@@ -1161,7 +1161,7 @@ sdadas__mmlw_roberta_base = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/sdadas/mmlw-roberta-base",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1169,7 +1169,7 @@ sdadas__mmlw_roberta_base = ModelMeta(
     adapted_from="sdadas/polish-roberta-base-v2",
     superseded_by=None,
     citation="""@article{dadas2024pirb,
-  title={{PIRB}: A Comprehensive Benchmark of Polish Dense and Hybrid Text Retrieval Methods}, 
+  title={{PIRB}: A Comprehensive Benchmark of Polish Dense and Hybrid Text Retrieval Methods},
   author={Sławomir Dadas and Michał Perełkiewicz and Rafał Poświata},
   year={2024},
   eprint={2402.13350},
@@ -1192,7 +1192,7 @@ sdadas__mmlw_roberta_large = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/sdadas/mmlw-roberta-large",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1200,7 +1200,7 @@ sdadas__mmlw_roberta_large = ModelMeta(
     adapted_from="sdadas/polish-roberta-large-v2",
     superseded_by=None,
     citation="""@article{dadas2024pirb,
-  title={{PIRB}: A Comprehensive Benchmark of Polish Dense and Hybrid Text Retrieval Methods}, 
+  title={{PIRB}: A Comprehensive Benchmark of Polish Dense and Hybrid Text Retrieval Methods},
   author={Sławomir Dadas and Michał Perełkiewicz and Rafał Poświata},
   year={2024},
   eprint={2402.13350},
@@ -1278,7 +1278,7 @@ izhx__udever_bloom_1b1 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/izhx/udever-bloom-1b1",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1307,7 +1307,7 @@ izhx__udever_bloom_3b = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/izhx/udever-bloom-3b",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1336,7 +1336,7 @@ izhx__udever_bloom_560m = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/izhx/udever-bloom-560m",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1365,7 +1365,7 @@ izhx__udever_bloom_7b1 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers"],
     reference="https://huggingface.co/izhx/udever-bloom-7b1",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1394,7 +1394,7 @@ avsolatorio__gist_embedding_v0 = 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/avsolatorio/GIST-Embedding-v0",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1444,7 +1444,7 @@ avsolatorio__gist_all_minilm_l6_v2 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/avsolatorio/GIST-all-MiniLM-L6-v2",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1494,7 +1494,7 @@ avsolatorio__gist_large_embedding_v0 = 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/avsolatorio/GIST-large-Embedding-v0",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1544,7 +1544,7 @@ avsolatorio__gist_small_embedding_v0 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/avsolatorio/GIST-small-Embedding-v0",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1594,7 +1594,7 @@ bigscience__sgpt_bloom_7b1_msmarco = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers"],
     reference="https://huggingface.co/bigscience/sgpt-bloom-7b1-msmarco",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1623,7 +1623,7 @@ aari1995__german_semantic_sts_v2 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors", "Transformers"],
     reference="https://huggingface.co/aari1995/German_Semantic_STS_V2",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1647,7 +1647,7 @@ abhinand__medembed_small_v0_1 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers", "safetensors"],
     reference="https://huggingface.co/abhinand/MedEmbed-small-v0.1",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1708,7 +1708,7 @@ brahmairesearch__slx_v0_1 = 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/brahmairesearch/slx-v0.1",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1731,7 +1731,7 @@ deepfile__embedder_100p = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Transformers", "safetensors"],
     reference="https://huggingface.co/deepfile/embedder-100p",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1754,7 +1754,7 @@ infgrad__stella_base_en_v2 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "Sentence Transformers"],
     reference="https://huggingface.co/infgrad/stella-base-en-v2",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1777,7 +1777,7 @@ malenia1__ternary_weight_embedding = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch"],
+    framework=["PyTorch", "safetensors"],
     reference="https://huggingface.co/malenia1/ternary-weight-embedding",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1800,7 +1800,7 @@ omarelshehy__arabic_english_sts_matryoshka = 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/omarelshehy/arabic-english-sts-matryoshka",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1840,7 +1840,7 @@ openbmb__minicpm_embedding = 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/openbmb/MiniCPM-Embedding",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1864,7 +1864,7 @@ silma_ai__silma_embedding_matryoshka_v0_1 = 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/silma-ai/silma-embeddding-matryoshka-v0.1",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1895,7 +1895,7 @@ sbert_chinese_general_v1 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "Transformers"],
     reference="https://huggingface.co/DMetaSoul/sbert-chinese-general-v1",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1923,7 +1923,7 @@ dmeta_embedding_zh_small = 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/DMetaSoul/Dmeta-embedding-zh-small/",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1946,7 +1946,7 @@ xiaobu_embedding = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "Transformers"],
     reference="https://huggingface.co/lier007/xiaobu-embedding",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1970,7 +1970,7 @@ xiaobu_embedding_v2 = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "ONNX", "safetensors"],
     reference="https://huggingface.co/lier007/xiaobu-embedding-v2",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -1994,7 +1994,7 @@ yinka_embedding = ModelMeta(
     open_weights=True,
     public_training_code=None,
     public_training_data=None,
-    framework=["PyTorch", "Sentence Transformers"],
+    framework=["PyTorch", "Sentence Transformers", "Transformers"],
     reference="https://huggingface.co/Classical/Yinka",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -2017,7 +2017,7 @@ conan_embedding = 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/Classical/Yinka",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
@@ -2025,13 +2025,13 @@ conan_embedding = ModelMeta(
     training_datasets=None,  # They "scraped" things from the internet, we don't know, could be leakage
     superseded_by=None,
     citation="""@misc{li2024conanembeddinggeneraltextembedding,
-  title={Conan-embedding: General Text Embedding with More and Better Negative Samples}, 
+  title={Conan-embedding: General Text Embedding with More and Better Negative Samples},
   author={Shiyu Li and Yang Tang and Shizhe Chen and Xi Chen},
   year={2024},
   eprint={2408.15710},
   archivePrefix={arXiv},
   primaryClass={cs.CL},
-  url={https://arxiv.org/abs/2408.15710}, 
+  url={https://arxiv.org/abs/2408.15710},
 }""",
 )
 
@@ -2050,14 +2050,14 @@ ember_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/llmrails/ember-v1",
     similarity_fn_name=ScoringFunction.COSINE,
     use_instructions=None,
     training_datasets=None,
     superseded_by=None,
     citation="""@misc{nur2024emberv1,
-      title={ember-v1: SOTA embedding model}, 
+      title={ember-v1: SOTA embedding model},
       author={Enrike Nur and Anar Aliyev},
       year={2023},
 }""",

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

@@ -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,
@@ -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,