Spaces:
Running
Running
BenchmarkBot
commited on
Commit
β’
02b17f3
1
Parent(s):
e2d1670
fix optimizations names
Browse files
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: "
|
|
|
|
|
|
|
|
|
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 |
|