Spaces:
Running
Running
Update data/model_handler.py
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
| 82 |
else:
|
| 83 |
-
dataset_res["Model Size (Million Parameters)"] =
|
| 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
|