djstrong commited on
Commit
a639a0d
1 Parent(s): 008f87f

chat tempalte evaluations

Browse files
Files changed (1) hide show
  1. src/leaderboard/read_evals.py +4 -0
src/leaderboard/read_evals.py CHANGED
@@ -46,6 +46,7 @@ class EvalResult:
46
  config = data.get("config")
47
  n_shot = data.get("n-shot")
48
  start_date = data.get("date", 0)
 
49
 
50
  # Precision
51
  precision = Precision.from_str(config.get("model_dtype"))
@@ -87,6 +88,9 @@ class EvalResult:
87
  result_key = f"{org}_{model}" # _{precision.value.name}
88
  full_model = "/".join(org_and_model)
89
 
 
 
 
90
  still_on_hub, err, model_config = is_model_on_hub(
91
  full_model.split(',')[0], config.get("model_sha", "main"), trust_remote_code=True, test_tokenizer=False
92
  )
 
46
  config = data.get("config")
47
  n_shot = data.get("n-shot")
48
  start_date = data.get("date", 0)
49
+ chat_template = data.get("chat_template", None)
50
 
51
  # Precision
52
  precision = Precision.from_str(config.get("model_dtype"))
 
88
  result_key = f"{org}_{model}" # _{precision.value.name}
89
  full_model = "/".join(org_and_model)
90
 
91
+ if chat_template:
92
+ result_key = f"{result_key}_chat"
93
+
94
  still_on_hub, err, model_config = is_model_on_hub(
95
  full_model.split(',')[0], config.get("model_sha", "main"), trust_remote_code=True, test_tokenizer=False
96
  )