t0-0 commited on
Commit
4263de1
1 Parent(s): 2c5b4cb

Update src/leaderboard/read_evals.py

Browse files
Files changed (1) hide show
  1. src/leaderboard/read_evals.py +6 -1
src/leaderboard/read_evals.py CHANGED
@@ -34,6 +34,7 @@ class EvalResult:
34
  date: str = "" # submission date of request file
35
  still_on_hub: bool = False
36
  num_few_shots: str = "0"
 
37
 
38
  @classmethod
39
  def init_from_json_file(self, json_filepath):
@@ -59,6 +60,9 @@ class EvalResult:
59
  precision = model_config.get("dtype", "Unknown")
60
  print("precision", precision)
61
 
 
 
 
62
  # Get model and org
63
  org_and_model = config.get("model_name", config.get("model").get("model", None))
64
  org_and_model = org_and_model.split("/", 1)
@@ -106,7 +110,8 @@ class EvalResult:
106
  still_on_hub=still_on_hub,
107
  architecture=architecture,
108
  model_type=model_type,
109
- num_few_shots=num_few_shots
 
110
  )
111
 
112
  def update_with_request_file(self, requests_path):
 
34
  date: str = "" # submission date of request file
35
  still_on_hub: bool = False
36
  num_few_shots: str = "0"
37
+ add_special_tokens = True
38
 
39
  @classmethod
40
  def init_from_json_file(self, json_filepath):
 
60
  precision = model_config.get("dtype", "Unknown")
61
  print("precision", precision)
62
 
63
+ # Add Special Tokens
64
+ add_special_tokens = config.get("pipeline_kwargs").get("add_special_tokens")
65
+
66
  # Get model and org
67
  org_and_model = config.get("model_name", config.get("model").get("model", None))
68
  org_and_model = org_and_model.split("/", 1)
 
110
  still_on_hub=still_on_hub,
111
  architecture=architecture,
112
  model_type=model_type,
113
+ num_few_shots=num_few_shots,
114
+ add_special_tokens=add_special_tokens,
115
  )
116
 
117
  def update_with_request_file(self, requests_path):