BenchmarkBot commited on
Commit
af2159b
β€’
1 Parent(s): f208a6d

added ort benchmarks

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -42,6 +42,9 @@ def get_benchmark_df(benchmark):
42
 
43
  # preprocess
44
  bench_df["model"] = bench_df["model"].apply(make_clickable_model)
 
 
 
45
  # filter
46
  bench_df = bench_df[list(COLUMNS_MAPPING.keys())]
47
  # rename
@@ -68,12 +71,6 @@ def change_tab(query_param):
68
 
69
  def submit_query(single_df, multi_df, text, backends, datatypes, threshold):
70
 
71
- print("submit_query")
72
- print(text)
73
- print(backends)
74
- print(datatypes)
75
- print(threshold)
76
-
77
  filtered_single = single_df[
78
  single_df["Model πŸ€—"].str.contains(text) &
79
  single_df["Backend 🏭"].isin(backends) &
@@ -105,7 +102,7 @@ with demo:
105
  )
106
  backend_checkboxes = gr.CheckboxGroup(
107
  choices=["pytorch", "onnxruntime"],
108
- value=["pytorch"],
109
  label="Backends 🏭",
110
  info="Select the backends",
111
  elem_id="backend-checkboxes",
 
42
 
43
  # preprocess
44
  bench_df["model"] = bench_df["model"].apply(make_clickable_model)
45
+ # set none datatype to float32
46
+ bench_df["backend.torch_dtype"] = bench_df["backend.torch_dtype"].fillna(
47
+ "float32")
48
  # filter
49
  bench_df = bench_df[list(COLUMNS_MAPPING.keys())]
50
  # rename
 
71
 
72
  def submit_query(single_df, multi_df, text, backends, datatypes, threshold):
73
 
 
 
 
 
 
 
74
  filtered_single = single_df[
75
  single_df["Model πŸ€—"].str.contains(text) &
76
  single_df["Backend 🏭"].isin(backends) &
 
102
  )
103
  backend_checkboxes = gr.CheckboxGroup(
104
  choices=["pytorch", "onnxruntime"],
105
+ value=["pytorch", "onnxruntime"],
106
  label="Backends 🏭",
107
  info="Select the backends",
108
  elem_id="backend-checkboxes",