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)
|