Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Tom Aarsen
commited on
Commit
·
a9153cc
1
Parent(s):
9bce65f
Update edge case where model is not specified
Browse files
app.py
CHANGED
|
@@ -258,7 +258,7 @@ def get_mteb_data(tasks=["Clustering"], langs=[], datasets=[], fillna=True, add_
|
|
| 258 |
EXTERNAL_MODEL_TO_DIM.get(name_without_org, ""),
|
| 259 |
EXTERNAL_MODEL_TO_SEQLEN.get(name_without_org, ""),
|
| 260 |
EXTERNAL_MODEL_TO_SIZE.get(name_without_org, ""),
|
| 261 |
-
round(EXTERNAL_MODEL_TO_SIZE
|
| 262 |
)
|
| 263 |
out["Embedding Dimensions"], out["Max Tokens"], out["Model Size (Million Parameters)"], out["Memory Usage (GB, fp32)"] = tuple(MODEL_INFOS[model.modelId]["dim_seq_size"])
|
| 264 |
df_list.append(out)
|
|
|
|
| 258 |
EXTERNAL_MODEL_TO_DIM.get(name_without_org, ""),
|
| 259 |
EXTERNAL_MODEL_TO_SEQLEN.get(name_without_org, ""),
|
| 260 |
EXTERNAL_MODEL_TO_SIZE.get(name_without_org, ""),
|
| 261 |
+
round(EXTERNAL_MODEL_TO_SIZE[name_without_org] * 1e6 * 4 / 1024**3, 2) if name_without_org in EXTERNAL_MODEL_TO_SIZE else "",
|
| 262 |
)
|
| 263 |
out["Embedding Dimensions"], out["Max Tokens"], out["Model Size (Million Parameters)"], out["Memory Usage (GB, fp32)"] = tuple(MODEL_INFOS[model.modelId]["dim_seq_size"])
|
| 264 |
df_list.append(out)
|