QuentinJG commited on
Commit
140b349
·
verified ·
1 Parent(s): 04989cc

Update data/model_handler.py

Browse files
Files changed (1) hide show
  1. data/model_handler.py +5 -2
data/model_handler.py CHANGED
@@ -78,9 +78,12 @@ class ModelHandler:
78
  dataset_res = {}
79
  keywords = VIDORE_V1_MTEB_NAMES if benchmark_version == 1 else VIDORE_V2_MTEB_NAMES
80
  if "n_parameters" in filtered_model_infos[model]["meta"]:
81
- dataset_res["Model Size (Million Parameters)"] = filtered_model_infos[model]["meta"]["n_parameters"] // 1_000_000
 
 
 
82
  else:
83
- dataset_res["Model Size (Million Parameters)"] = None
84
  for dataset in res.keys():
85
  if not any(keyword in dataset for keyword in keywords):
86
  continue
 
78
  dataset_res = {}
79
  keywords = VIDORE_V1_MTEB_NAMES if benchmark_version == 1 else VIDORE_V2_MTEB_NAMES
80
  if "n_parameters" in filtered_model_infos[model]["meta"]:
81
+ try:
82
+ dataset_res["Model Size (Million Parameters)"] = filtered_model_infos[model]["meta"]["n_parameters"] // 1_000_000
83
+ except TypeError:
84
+ dataset_res["Model Size (Million Parameters)"] = -1
85
  else:
86
+ dataset_res["Model Size (Million Parameters)"] = -1
87
  for dataset in res.keys():
88
  if not any(keyword in dataset for keyword in keywords):
89
  continue