BenchmarkBot commited on
Commit
02b17f3
β€’
1 Parent(s): e2d1670

fix optimizations names

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -67,7 +67,11 @@ def get_benchmark_df(benchmark="1xA100-80GB"):
67
  bench_df["optimizations"] = bench_df[
68
  ["backend.bettertransformer", "backend.load_in_8bit", "backend.load_in_4bit"]
69
  ].apply(
70
- lambda x: ", ".join([opt for opt in x.index if x[opt] == True]),
 
 
 
 
71
  axis=1,
72
  )
73
 
 
67
  bench_df["optimizations"] = bench_df[
68
  ["backend.bettertransformer", "backend.load_in_8bit", "backend.load_in_4bit"]
69
  ].apply(
70
+ lambda x: "BetterTransformer"
71
+ if x[0] == True
72
+ else (
73
+ "LLM.int8" if x[1] == True else ("NF4" if x[2] == True else "")
74
+ ),
75
  axis=1,
76
  )
77