Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Tom Aarsen
commited on
Commit
·
a8ba8f1
1
Parent(s):
0ebd4b8
Move globals around slightly
Browse files
app.py
CHANGED
|
@@ -946,6 +946,104 @@ PROPRIETARY_MODELS = {
|
|
| 946 |
for model in PROPRIETARY_MODELS
|
| 947 |
}
|
| 948 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 949 |
MODELS_TO_SKIP = {
|
| 950 |
"baseplate/instructor-large-1", # Duplicate
|
| 951 |
"radames/e5-large", # Duplicate
|
|
@@ -1070,103 +1168,6 @@ MODELS_TO_SKIP = {
|
|
| 1070 |
"Koat/gte-tiny",
|
| 1071 |
}
|
| 1072 |
|
| 1073 |
-
SENTENCE_TRANSFORMERS_COMPATIBLE_MODELS = {
|
| 1074 |
-
"allenai-specter",
|
| 1075 |
-
"allenai-specter",
|
| 1076 |
-
"all-MiniLM-L12-v2",
|
| 1077 |
-
"all-MiniLM-L6-v2",
|
| 1078 |
-
"all-mpnet-base-v2",
|
| 1079 |
-
"bert-base-10lang-cased",
|
| 1080 |
-
"bert-base-15lang-cased",
|
| 1081 |
-
"bert-base-25lang-cased",
|
| 1082 |
-
"bert-base-multilingual-cased",
|
| 1083 |
-
"bert-base-multilingual-uncased",
|
| 1084 |
-
"bert-base-swedish-cased",
|
| 1085 |
-
"bert-base-uncased",
|
| 1086 |
-
"bge-base-zh-v1.5",
|
| 1087 |
-
"bge-large-zh-v1.5",
|
| 1088 |
-
"bge-large-zh-noinstruct",
|
| 1089 |
-
"bge-small-zh-v1.5",
|
| 1090 |
-
"camembert-base",
|
| 1091 |
-
"camembert-large",
|
| 1092 |
-
"contriever-base-msmarco",
|
| 1093 |
-
"cross-en-de-roberta-sentence-transformer",
|
| 1094 |
-
"DanskBERT",
|
| 1095 |
-
"distilbert-base-25lang-cased",
|
| 1096 |
-
"distilbert-base-en-fr-cased",
|
| 1097 |
-
"distilbert-base-en-fr-es-pt-it-cased",
|
| 1098 |
-
"distilbert-base-fr-cased",
|
| 1099 |
-
"distilbert-base-uncased",
|
| 1100 |
-
"distiluse-base-multilingual-cased-v2",
|
| 1101 |
-
"dfm-encoder-large-v1",
|
| 1102 |
-
"dfm-sentence-encoder-large-1",
|
| 1103 |
-
"e5-base",
|
| 1104 |
-
"e5-large",
|
| 1105 |
-
"e5-mistral-7b-instruct",
|
| 1106 |
-
"e5-small",
|
| 1107 |
-
"electra-small-nordic",
|
| 1108 |
-
"electra-small-swedish-cased-discriminator",
|
| 1109 |
-
"flaubert_base_cased",
|
| 1110 |
-
"flaubert_base_uncased",
|
| 1111 |
-
"flaubert_large_cased",
|
| 1112 |
-
"gbert-base",
|
| 1113 |
-
"gbert-large",
|
| 1114 |
-
"gelectra-base",
|
| 1115 |
-
"gelectra-large",
|
| 1116 |
-
"glove.6B.300d",
|
| 1117 |
-
"gottbert-base",
|
| 1118 |
-
"gtr-t5-base",
|
| 1119 |
-
"gtr-t5-large",
|
| 1120 |
-
"gtr-t5-xl",
|
| 1121 |
-
"gtr-t5-xxl",
|
| 1122 |
-
"herbert-base-retrieval-v2",
|
| 1123 |
-
"komninos",
|
| 1124 |
-
"luotuo-bert-medium",
|
| 1125 |
-
"LaBSE",
|
| 1126 |
-
"m3e-base",
|
| 1127 |
-
"m3e-large",
|
| 1128 |
-
"msmarco-bert-co-condensor",
|
| 1129 |
-
"multi-qa-MiniLM-L6-cos-v1",
|
| 1130 |
-
"multilingual-e5-base",
|
| 1131 |
-
"multilingual-e5-large",
|
| 1132 |
-
"multilingual-e5-small",
|
| 1133 |
-
"nb-bert-base",
|
| 1134 |
-
"nb-bert-large",
|
| 1135 |
-
"nomic-embed-text-v1.5-64",
|
| 1136 |
-
"nomic-embed-text-v1.5-128",
|
| 1137 |
-
"nomic-embed-text-v1.5-256",
|
| 1138 |
-
"nomic-embed-text-v1.5-512",
|
| 1139 |
-
"norbert3-base",
|
| 1140 |
-
"norbert3-large",
|
| 1141 |
-
"paraphrase-multilingual-mpnet-base-v2",
|
| 1142 |
-
"paraphrase-multilingual-MiniLM-L12-v2",
|
| 1143 |
-
"sentence-camembert-base",
|
| 1144 |
-
"sentence-camembert-large",
|
| 1145 |
-
"sentence-croissant-llm-base",
|
| 1146 |
-
"sentence-bert-swedish-cased",
|
| 1147 |
-
"sentence-t5-base",
|
| 1148 |
-
"sentence-t5-large",
|
| 1149 |
-
"sentence-t5-xl",
|
| 1150 |
-
"sentence-t5-xxl",
|
| 1151 |
-
"silver-retriever-base-v1",
|
| 1152 |
-
"sup-simcse-bert-base-uncased",
|
| 1153 |
-
"st-polish-paraphrase-from-distilroberta",
|
| 1154 |
-
"st-polish-paraphrase-from-mpnet",
|
| 1155 |
-
"text2vec-base-chinese",
|
| 1156 |
-
"text2vec-large-chinese",
|
| 1157 |
-
"udever-bloom-1b1",
|
| 1158 |
-
"udever-bloom-560m",
|
| 1159 |
-
"universal-sentence-encoder-multilingual-3",
|
| 1160 |
-
"universal-sentence-encoder-multilingual-large-3",
|
| 1161 |
-
"unsup-simcse-bert-base-uncased",
|
| 1162 |
-
"use-cmlm-multilingual",
|
| 1163 |
-
"xlm-roberta-base",
|
| 1164 |
-
"xlm-roberta-large",
|
| 1165 |
-
}
|
| 1166 |
-
SENTENCE_TRANSFORMERS_COMPATIBLE_MODELS = {
|
| 1167 |
-
make_clickable_model(model, link=EXTERNAL_MODEL_TO_LINK.get(model, "https://huggingface.co/spaces/mteb/leaderboard"))
|
| 1168 |
-
for model in SENTENCE_TRANSFORMERS_COMPATIBLE_MODELS
|
| 1169 |
-
}
|
| 1170 |
|
| 1171 |
def add_lang(examples):
|
| 1172 |
if not(examples["eval_language"]):
|
|
|
|
| 946 |
for model in PROPRIETARY_MODELS
|
| 947 |
}
|
| 948 |
|
| 949 |
+
SENTENCE_TRANSFORMERS_COMPATIBLE_MODELS = {
|
| 950 |
+
"allenai-specter",
|
| 951 |
+
"allenai-specter",
|
| 952 |
+
"all-MiniLM-L12-v2",
|
| 953 |
+
"all-MiniLM-L6-v2",
|
| 954 |
+
"all-mpnet-base-v2",
|
| 955 |
+
"bert-base-10lang-cased",
|
| 956 |
+
"bert-base-15lang-cased",
|
| 957 |
+
"bert-base-25lang-cased",
|
| 958 |
+
"bert-base-multilingual-cased",
|
| 959 |
+
"bert-base-multilingual-uncased",
|
| 960 |
+
"bert-base-swedish-cased",
|
| 961 |
+
"bert-base-uncased",
|
| 962 |
+
"bge-base-zh-v1.5",
|
| 963 |
+
"bge-large-zh-v1.5",
|
| 964 |
+
"bge-large-zh-noinstruct",
|
| 965 |
+
"bge-small-zh-v1.5",
|
| 966 |
+
"camembert-base",
|
| 967 |
+
"camembert-large",
|
| 968 |
+
"contriever-base-msmarco",
|
| 969 |
+
"cross-en-de-roberta-sentence-transformer",
|
| 970 |
+
"DanskBERT",
|
| 971 |
+
"distilbert-base-25lang-cased",
|
| 972 |
+
"distilbert-base-en-fr-cased",
|
| 973 |
+
"distilbert-base-en-fr-es-pt-it-cased",
|
| 974 |
+
"distilbert-base-fr-cased",
|
| 975 |
+
"distilbert-base-uncased",
|
| 976 |
+
"distiluse-base-multilingual-cased-v2",
|
| 977 |
+
"dfm-encoder-large-v1",
|
| 978 |
+
"dfm-sentence-encoder-large-1",
|
| 979 |
+
"e5-base",
|
| 980 |
+
"e5-large",
|
| 981 |
+
"e5-mistral-7b-instruct",
|
| 982 |
+
"e5-small",
|
| 983 |
+
"electra-small-nordic",
|
| 984 |
+
"electra-small-swedish-cased-discriminator",
|
| 985 |
+
"flaubert_base_cased",
|
| 986 |
+
"flaubert_base_uncased",
|
| 987 |
+
"flaubert_large_cased",
|
| 988 |
+
"gbert-base",
|
| 989 |
+
"gbert-large",
|
| 990 |
+
"gelectra-base",
|
| 991 |
+
"gelectra-large",
|
| 992 |
+
"glove.6B.300d",
|
| 993 |
+
"gottbert-base",
|
| 994 |
+
"gtr-t5-base",
|
| 995 |
+
"gtr-t5-large",
|
| 996 |
+
"gtr-t5-xl",
|
| 997 |
+
"gtr-t5-xxl",
|
| 998 |
+
"herbert-base-retrieval-v2",
|
| 999 |
+
"komninos",
|
| 1000 |
+
"luotuo-bert-medium",
|
| 1001 |
+
"LaBSE",
|
| 1002 |
+
"m3e-base",
|
| 1003 |
+
"m3e-large",
|
| 1004 |
+
"msmarco-bert-co-condensor",
|
| 1005 |
+
"multi-qa-MiniLM-L6-cos-v1",
|
| 1006 |
+
"multilingual-e5-base",
|
| 1007 |
+
"multilingual-e5-large",
|
| 1008 |
+
"multilingual-e5-small",
|
| 1009 |
+
"nb-bert-base",
|
| 1010 |
+
"nb-bert-large",
|
| 1011 |
+
"nomic-embed-text-v1.5-64",
|
| 1012 |
+
"nomic-embed-text-v1.5-128",
|
| 1013 |
+
"nomic-embed-text-v1.5-256",
|
| 1014 |
+
"nomic-embed-text-v1.5-512",
|
| 1015 |
+
"norbert3-base",
|
| 1016 |
+
"norbert3-large",
|
| 1017 |
+
"paraphrase-multilingual-mpnet-base-v2",
|
| 1018 |
+
"paraphrase-multilingual-MiniLM-L12-v2",
|
| 1019 |
+
"sentence-camembert-base",
|
| 1020 |
+
"sentence-camembert-large",
|
| 1021 |
+
"sentence-croissant-llm-base",
|
| 1022 |
+
"sentence-bert-swedish-cased",
|
| 1023 |
+
"sentence-t5-base",
|
| 1024 |
+
"sentence-t5-large",
|
| 1025 |
+
"sentence-t5-xl",
|
| 1026 |
+
"sentence-t5-xxl",
|
| 1027 |
+
"silver-retriever-base-v1",
|
| 1028 |
+
"sup-simcse-bert-base-uncased",
|
| 1029 |
+
"st-polish-paraphrase-from-distilroberta",
|
| 1030 |
+
"st-polish-paraphrase-from-mpnet",
|
| 1031 |
+
"text2vec-base-chinese",
|
| 1032 |
+
"text2vec-large-chinese",
|
| 1033 |
+
"udever-bloom-1b1",
|
| 1034 |
+
"udever-bloom-560m",
|
| 1035 |
+
"universal-sentence-encoder-multilingual-3",
|
| 1036 |
+
"universal-sentence-encoder-multilingual-large-3",
|
| 1037 |
+
"unsup-simcse-bert-base-uncased",
|
| 1038 |
+
"use-cmlm-multilingual",
|
| 1039 |
+
"xlm-roberta-base",
|
| 1040 |
+
"xlm-roberta-large",
|
| 1041 |
+
}
|
| 1042 |
+
SENTENCE_TRANSFORMERS_COMPATIBLE_MODELS = {
|
| 1043 |
+
make_clickable_model(model, link=EXTERNAL_MODEL_TO_LINK.get(model, "https://huggingface.co/spaces/mteb/leaderboard"))
|
| 1044 |
+
for model in SENTENCE_TRANSFORMERS_COMPATIBLE_MODELS
|
| 1045 |
+
}
|
| 1046 |
+
|
| 1047 |
MODELS_TO_SKIP = {
|
| 1048 |
"baseplate/instructor-large-1", # Duplicate
|
| 1049 |
"radames/e5-large", # Duplicate
|
|
|
|
| 1168 |
"Koat/gte-tiny",
|
| 1169 |
}
|
| 1170 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1171 |
|
| 1172 |
def add_lang(examples):
|
| 1173 |
if not(examples["eval_language"]):
|