t0-0 commited on
Commit
15daadb
1 Parent(s): abbce0b

Update src/leaderboard/read_evals.py

Browse files
Files changed (1) hide show
  1. src/leaderboard/read_evals.py +2 -2
src/leaderboard/read_evals.py CHANGED
@@ -59,10 +59,10 @@ class EvalResult:
59
  precision = model_config.get("dtype", "Unknown")
60
 
61
  # Add Special Tokens
62
- add_special_tokens = str(config.get("pipeline_kwargs").get("add_special_tokens"))
63
 
64
  # Get model and org
65
- org_and_model = config.get("model_name", config.get("model").get("model", None))
66
  org_and_model = org_and_model.split("/", 1)
67
 
68
  if len(org_and_model) == 1:
 
59
  precision = model_config.get("dtype", "Unknown")
60
 
61
  # Add Special Tokens
62
+ add_special_tokens = str(config.get("pipeline_kwargs",{"add_special_tokens":"Unknown"}).get("add_special_tokens"))
63
 
64
  # Get model and org
65
+ org_and_model = config.get("model_name", config.get("offline_inference").get("model_name", None))
66
  org_and_model = org_and_model.split("/", 1)
67
 
68
  if len(org_and_model) == 1: