Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Merge pull request #18 from Samoed/fix_leaderboard
Browse files- refresh.py +2 -1
refresh.py
CHANGED
@@ -300,7 +300,8 @@ def get_mteb_data(tasks=["Clustering"], langs=[], datasets=[], fillna=True, add_
|
|
300 |
)
|
301 |
out["Embedding Dimensions"], out["Max Tokens"], out["Model Size (Million Parameters)"], out["Memory Usage (GB, fp32)"] = tuple(MODEL_INFOS[model.modelId]["dim_seq_size"])
|
302 |
df_list.append(out)
|
303 |
-
|
|
|
304 |
SENTENCE_TRANSFORMERS_COMPATIBLE_MODELS.add(out["Model"])
|
305 |
|
306 |
# # Save & cache MODEL_INFOS
|
|
|
300 |
)
|
301 |
out["Embedding Dimensions"], out["Max Tokens"], out["Model Size (Million Parameters)"], out["Memory Usage (GB, fp32)"] = tuple(MODEL_INFOS[model.modelId]["dim_seq_size"])
|
302 |
df_list.append(out)
|
303 |
+
model_siblings = model.siblings or []
|
304 |
+
if model.library_name == "sentence-transformers" or "sentence-transformers" in model.tags or "modules.json" in {file.rfilename for file in model_siblings}:
|
305 |
SENTENCE_TRANSFORMERS_COMPATIBLE_MODELS.add(out["Model"])
|
306 |
|
307 |
# # Save & cache MODEL_INFOS
|