Spaces:
Running
Running
BenchmarkBot
commited on
Commit
β’
f41da95
1
Parent(s):
aa6a202
reduce weight class and model type sizes
Browse files
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": "
|
33 |
-
"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["
|
111 |
-
bench_df["
|
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(
|