add peak memory slider

app.py

@@ -150,7 +150,13 @@ def get_benchmark_plot(bench_df):
 
 
 def filter_query(
-    text, backends, datatypes, optimizations, threshold, benchmark="1xA100-80GB"
+    text,
+    backends,
+    datatypes,
+    optimizations,
+    score,
+    memory,
+    benchmark="1xA100-80GB",
 ):
     raw_df = get_benchmark_df(benchmark=benchmark)
 
@@ -169,7 +175,8 @@ def filter_query(
             if len(optimizations) > 0
             else True
         )
-        & (raw_df["average"] >= threshold)
+        & (raw_df["average"] >= score)
+        & (raw_df["forward.peak_memory(MB)"] <= memory)
     ]
 
     filtered_table = get_benchmark_table(filtered_df)
@@ -197,19 +204,25 @@ with demo:
 
     # control panel interface
     with gr.Row():
-        with gr.Column(scale=1):
-            search_bar = gr.Textbox(
-                label="Model 🤗",
-                info="🔍 Search for a model name",
-                elem_id="search-bar",
-            )
-        with gr.Column(scale=1):
-            score_slider = gr.Slider(
-                label="Average Open LLM Score 📈",
-                info="🎚️ Slide to minimum Average Open LLM score",
-                value=0.0,
-                elem_id="threshold-slider",
-            )
+        search_bar = gr.Textbox(
+            label="Model 🤗",
+            info="🔍 Search for a model name",
+            elem_id="search-bar",
+        )
+        score_slider = gr.Slider(
+            label="Average Open LLM Score 📈",
+            info="🎚️ Slide to minimum Average Open LLM score",
+            value=0,
+            elem_id="threshold-slider",
+        )
+        memory_slider = gr.Slider(
+            label="Peak Memory (MB) 📈",
+            info="🎚️ Slide to maximum Peak Memory",
+            minimum=0,
+            maximum=80 * 1024,
+            value=80 * 1024,
+            elem_id="memory-slider",
+        )
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -275,6 +288,7 @@ with demo:
                 datatype_checkboxes,
                 optimizations_checkboxes,
                 score_slider,
+                memory_slider,
             ],
             [single_A100_leaderboard, single_A100_plotly],
         )