Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
eduagarcia
commited on
Commit
•
349b10b
1
Parent(s):
bbfe97c
Fix a weird bug that made the cicklabe model name fail to render in some boards
Browse files
app.py
CHANGED
@@ -221,7 +221,7 @@ def get_mteb_data(tasks=["Clustering"], langs=[], datasets=[], fillna=True, add_
|
|
221 |
MODEL_EMB_DIM[model.modelId] = get_dim_seq_size(model)
|
222 |
out["Embedding Dimensions"], out["Max Tokens"], out["Model Size (Million Parameters)"], out["Memory Usage (GB, fp32)"] = MODEL_EMB_DIM[model.modelId]
|
223 |
except:
|
224 |
-
MODEL_EMB_DIM[model.modelId] =
|
225 |
df_list.append(out)
|
226 |
if model.library_name == "sentence-transformers" or "sentence-transformers" in model.tags or "modules.json" in {file.rfilename for file in model.siblings}:
|
227 |
SENTENCE_TRANSFORMERS_COMPATIBLE_MODELS.add(out["Model"])
|
@@ -294,11 +294,7 @@ for board, board_config in BOARDS_CONFIG.items():
|
|
294 |
all_data_tasks.extend(data_tasks.values())
|
295 |
else:
|
296 |
for task_category, task_category_list in board_config["tasks"].items():
|
297 |
-
|
298 |
-
if len(task_category_list) > 1:
|
299 |
-
columns.append("Average")
|
300 |
-
print(board, task_category, task_category_list)
|
301 |
-
data_task_category = get_mteb_data([task_category], [], task_category_list, refresh=False)[columns + task_category_list]
|
302 |
boards_data[board]["data_tasks"][task_category] = data_task_category
|
303 |
all_data_tasks.append(data_task_category)
|
304 |
|
@@ -388,7 +384,7 @@ for board, board_config in BOARDS_CONFIG.items():
|
|
388 |
"language_long": board_config["language_long"],
|
389 |
"description": f"**{task_category} {board_pretty_name}** {task_icon}{board_icon}",
|
390 |
"data": boards_data[board]["data_tasks"][task_category],
|
391 |
-
"refresh": partial(get_mteb_data, [task_category],
|
392 |
"credits": credits,
|
393 |
})
|
394 |
|
|
|
221 |
MODEL_EMB_DIM[model.modelId] = get_dim_seq_size(model)
|
222 |
out["Embedding Dimensions"], out["Max Tokens"], out["Model Size (Million Parameters)"], out["Memory Usage (GB, fp32)"] = MODEL_EMB_DIM[model.modelId]
|
223 |
except:
|
224 |
+
MODEL_EMB_DIM[model.modelId] = "", "", "", ""
|
225 |
df_list.append(out)
|
226 |
if model.library_name == "sentence-transformers" or "sentence-transformers" in model.tags or "modules.json" in {file.rfilename for file in model.siblings}:
|
227 |
SENTENCE_TRANSFORMERS_COMPATIBLE_MODELS.add(out["Model"])
|
|
|
294 |
all_data_tasks.extend(data_tasks.values())
|
295 |
else:
|
296 |
for task_category, task_category_list in board_config["tasks"].items():
|
297 |
+
data_task_category = get_mteb_data(tasks=[task_category], datasets=task_category_list, refresh=False)
|
|
|
|
|
|
|
|
|
298 |
boards_data[board]["data_tasks"][task_category] = data_task_category
|
299 |
all_data_tasks.append(data_task_category)
|
300 |
|
|
|
384 |
"language_long": board_config["language_long"],
|
385 |
"description": f"**{task_category} {board_pretty_name}** {task_icon}{board_icon}",
|
386 |
"data": boards_data[board]["data_tasks"][task_category],
|
387 |
+
"refresh": partial(get_mteb_data, tasks=[task_category], datasets=task_category_list),
|
388 |
"credits": credits,
|
389 |
})
|
390 |
|