Spaces:
Running
Running
Commit
β’
e8165e3
1
Parent(s):
df7d7c4
fix
Browse files
app.py
CHANGED
@@ -89,7 +89,6 @@ def get_benchmark_df(benchmark="Succeeded-1xA100-80GB"):
|
|
89 |
].fillna(1)
|
90 |
# fix nan values
|
91 |
merged_df[merged_df["generate.energy_consumption(tokens/kWh)"] == 1] = "N/A"
|
92 |
-
|
93 |
# add optimizations
|
94 |
merged_df["optimizations"] = merged_df["backend.bettertransformer"].apply(
|
95 |
lambda x: "BetterTransformer" if x else "None"
|
@@ -102,7 +101,7 @@ def get_benchmark_df(benchmark="Succeeded-1xA100-80GB"):
|
|
102 |
merged_df.sort_values(by=SORTING_COLUMN, ascending=SORTING_ASCENDING, inplace=True)
|
103 |
# drop duplicates
|
104 |
merged_df.drop_duplicates(subset=NO_DUPLICATES_COLUMNS, inplace=True)
|
105 |
-
merged_df = merged_df[[*
|
106 |
return merged_df
|
107 |
|
108 |
|
|
|
89 |
].fillna(1)
|
90 |
# fix nan values
|
91 |
merged_df[merged_df["generate.energy_consumption(tokens/kWh)"] == 1] = "N/A"
|
|
|
92 |
# add optimizations
|
93 |
merged_df["optimizations"] = merged_df["backend.bettertransformer"].apply(
|
94 |
lambda x: "BetterTransformer" if x else "None"
|
|
|
101 |
merged_df.sort_values(by=SORTING_COLUMN, ascending=SORTING_ASCENDING, inplace=True)
|
102 |
# drop duplicates
|
103 |
merged_df.drop_duplicates(subset=NO_DUPLICATES_COLUMNS, inplace=True)
|
104 |
+
merged_df = merged_df[[*list(ALL_COLUMNS_MAPPING.keys()), "generate.latency(s)"]]
|
105 |
return merged_df
|
106 |
|
107 |
|