Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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 |
|