BenchmarkBot commited on
Commit
f41da95
β€’
1 Parent(s): aa6a202

reduce weight class and model type sizes

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -29,8 +29,8 @@ OPTIMUM_TOKEN = os.environ.get("OPTIMUM_TOKEN", None)
29
 
30
  ALL_COLUMNS_MAPPING = {
31
  "best_scored_model": "Best Scored Model πŸ†",
32
- "model_type": "Model Type πŸ€—",
33
- "weight_class": "Weight Class πŸ‹οΈ",
34
  #
35
  "backend.name": "Backend 🏭",
36
  "backend.torch_dtype": "Dtype πŸ“₯",
@@ -107,8 +107,8 @@ def get_benchmark_table(bench_df):
107
  # rename
108
  bench_df.rename(columns=ALL_COLUMNS_MAPPING, inplace=True)
109
  # transform
110
- bench_df["Model Type πŸ€—"] = bench_df["Model Type πŸ€—"].apply(process_model_type)
111
- bench_df["Weight Class πŸ‹οΈ"] = bench_df["Weight Class πŸ‹οΈ"].apply(
112
  process_weight_class
113
  )
114
  bench_df["Best Scored Model πŸ†"] = bench_df["Best Scored Model πŸ†"].apply(
 
29
 
30
  ALL_COLUMNS_MAPPING = {
31
  "best_scored_model": "Best Scored Model πŸ†",
32
+ "model_type": "Type πŸ€—",
33
+ "weight_class": "Class πŸ‹οΈ",
34
  #
35
  "backend.name": "Backend 🏭",
36
  "backend.torch_dtype": "Dtype πŸ“₯",
 
107
  # rename
108
  bench_df.rename(columns=ALL_COLUMNS_MAPPING, inplace=True)
109
  # transform
110
+ bench_df["Type πŸ€—"] = bench_df["Type πŸ€—"].apply(process_model_type)
111
+ bench_df["Class πŸ‹οΈ"] = bench_df["Class πŸ‹οΈ"].apply(
112
  process_weight_class
113
  )
114
  bench_df["Best Scored Model πŸ†"] = bench_df["Best Scored Model πŸ†"].apply(