Commit
·
c944a42
1
Parent(s):
9072ddc
fix backend bug
Browse files
app.py
CHANGED
@@ -36,8 +36,9 @@ def get_benchmark_df(benchmark):
|
|
36 |
f"./llm-perf-dataset/reports/{benchmark}/inference_report.csv")
|
37 |
# preprocess
|
38 |
df["model"] = df["model"].apply(make_clickable_model)
|
39 |
-
df["backend"] = df["backend"].str.replace("pytorch", "Pytorch")
|
40 |
-
df["backend"] = df["backend"].str.replace(
|
|
|
41 |
# filter
|
42 |
df = df[list(COLUMNS_MAPPING.keys())]
|
43 |
# rename
|
|
|
36 |
f"./llm-perf-dataset/reports/{benchmark}/inference_report.csv")
|
37 |
# preprocess
|
38 |
df["model"] = df["model"].apply(make_clickable_model)
|
39 |
+
df["backend.name"] = df["backend.name"].str.replace("pytorch", "Pytorch")
|
40 |
+
df["backend.name"] = df["backend.name"].str.replace(
|
41 |
+
"onnxruntime", "OnnxRuntime")
|
42 |
# filter
|
43 |
df = df[list(COLUMNS_MAPPING.keys())]
|
44 |
# rename
|