sh1gechan commited on
Commit
c769424
·
verified ·
1 Parent(s): a718a71

Update src/leaderboard/read_evals.py

Browse files
Files changed (1) hide show
  1. src/leaderboard/read_evals.py +2 -1
src/leaderboard/read_evals.py CHANGED
@@ -67,7 +67,8 @@ class EvalResult:
67
  add_special_tokens = str(config.get("pipeline_kwargs",{"add_special_tokens":"Unknown"}).get("add_special_tokens"))
68
 
69
  # Get model and org
70
- org_and_model = config.get("model_name", config.get("offline_inference").get("model_name", None))
 
71
  org_and_model = org_and_model.split("/", 1)
72
 
73
  if len(org_and_model) == 1:
 
67
  add_special_tokens = str(config.get("pipeline_kwargs",{"add_special_tokens":"Unknown"}).get("add_special_tokens"))
68
 
69
  # Get model and org
70
+ # org_and_model = config.get("model_name", config.get("offline_inference").get("model_name", None))
71
+ org_and_model = config.get("model_name", config.get("offline_inference", {}).get("model_name", "Unknown"))
72
  org_and_model = org_and_model.split("/", 1)
73
 
74
  if len(org_and_model) == 1: