Orion Weller commited on
Commit
739bfda
2 Parent(s): f45efd6 a90ed79

Merge pull request #18 from Samoed/fix_leaderboard

Browse files
Files changed (1) hide show
  1. 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
- if model.library_name == "sentence-transformers" or "sentence-transformers" in model.tags or "modules.json" in {file.rfilename for file in model.siblings}:
 
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