Spaces:
Running
Running
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 |
|