XufengDuan commited on
Commit
f78bd7e
1 Parent(s): 9271c65

update scripts

Browse files
Files changed (2) hide show
  1. src/display/utils.py +6 -2
  2. src/populate.py +1 -1
src/display/utils.py CHANGED
@@ -29,7 +29,11 @@ auto_eval_column_dict.append(["model_type_symbol", ColumnContent,
29
  auto_eval_column_dict.append(["model", ColumnContent,
30
  ColumnContent("Model", "markdown", True, never_hidden=True)])
31
  for task in Tasks:
32
- auto_eval_column_dict.append([task.name, ColumnContent, ColumnContent(task.value.col_name, "number", True)])
 
 
 
 
33
  #
34
  # Model information
35
  auto_eval_column_dict.append(["model_type", ColumnContent, ColumnContent("Type", "str", False)])
@@ -95,7 +99,7 @@ class WeightType(Enum):
95
 
96
  class Precision(Enum):
97
  float16 = ModelDetails("float16")
98
- bfloat16 = ModelDetails("bfloat16")
99
  qt_8bit = ModelDetails("8bit")
100
  qt_4bit = ModelDetails("4bit")
101
  qt_GPTQ = ModelDetails("GPTQ")
 
29
  auto_eval_column_dict.append(["model", ColumnContent,
30
  ColumnContent("Model", "markdown", True, never_hidden=True)])
31
  for task in Tasks:
32
+ if task.value.col_name == "Overall Humanlike %" or task.value.col_name == "Overall CI":
33
+ auto_eval_column_dict.append([task.name, ColumnContent, ColumnContent(task.value.col_name, "number", True)])
34
+ else:
35
+ auto_eval_column_dict.append([task.name, ColumnContent, ColumnContent(task.value.col_name, "number", False)])
36
+
37
  #
38
  # Model information
39
  auto_eval_column_dict.append(["model_type", ColumnContent, ColumnContent("Type", "str", False)])
 
99
 
100
  class Precision(Enum):
101
  float16 = ModelDetails("float16")
102
+ bfloat16 = ModelDetails("BF16")
103
  qt_8bit = ModelDetails("8bit")
104
  qt_4bit = ModelDetails("4bit")
105
  qt_GPTQ = ModelDetails("GPTQ")
src/populate.py CHANGED
@@ -16,7 +16,7 @@ def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchm
16
 
17
  #print(all_data_json)
18
  df = pd.DataFrame.from_records(all_data_json)
19
- print("all results:",df)
20
  # exit()
21
  try:
22
  df = df.sort_values(by=[utils.AutoEvalColumn.hallucination_rate.name], ascending=True)
 
16
 
17
  #print(all_data_json)
18
  df = pd.DataFrame.from_records(all_data_json)
19
+ print("all results:",df.to_string())
20
  # exit()
21
  try:
22
  df = df.sort_values(by=[utils.AutoEvalColumn.hallucination_rate.name], ascending=True)