Tom Aarsen commited on
Commit
a9153cc
1 Parent(s): 9bce65f

Update edge case where model is not specified

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.get(name_without_org, "") * 1e6 * 4 / 1024**3, 2),
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)