Spaces:
Running
Running
Commit
ยท
affd732
1
Parent(s):
2460b35
fix filters
Browse files- src/control_panel.py +5 -0
src/control_panel.py
CHANGED
@@ -5,6 +5,7 @@ from src.leaderboard import get_leaderboard_df
|
|
5 |
from src.latency_score_memory import get_lat_score_mem_fig
|
6 |
from src.bettertransformer import get_bt_prefill_fig, get_bt_decode_fig
|
7 |
from src.flashattentionv2 import get_fa2_prefill_fig, get_fa2_decode_fig
|
|
|
8 |
|
9 |
|
10 |
def create_control_panel(machine: str = "hf-dgx-01"):
|
@@ -114,6 +115,8 @@ def filter_fn(
|
|
114 |
filtered_bt_decode_fig = get_bt_decode_fig(filtered_df)
|
115 |
filtered_fa2_prefill_fig = get_fa2_prefill_fig(filtered_df)
|
116 |
filtered_fa2_decode_fig = get_fa2_decode_fig(filtered_df)
|
|
|
|
|
117 |
|
118 |
return [
|
119 |
filtered_leaderboard_df,
|
@@ -122,6 +125,8 @@ def filter_fn(
|
|
122 |
filtered_bt_decode_fig,
|
123 |
filtered_fa2_prefill_fig,
|
124 |
filtered_fa2_decode_fig,
|
|
|
|
|
125 |
]
|
126 |
|
127 |
|
|
|
5 |
from src.latency_score_memory import get_lat_score_mem_fig
|
6 |
from src.bettertransformer import get_bt_prefill_fig, get_bt_decode_fig
|
7 |
from src.flashattentionv2 import get_fa2_prefill_fig, get_fa2_decode_fig
|
8 |
+
from src.exllama import get_exllama_prefill_fig, get_exllama_decode_fig
|
9 |
|
10 |
|
11 |
def create_control_panel(machine: str = "hf-dgx-01"):
|
|
|
115 |
filtered_bt_decode_fig = get_bt_decode_fig(filtered_df)
|
116 |
filtered_fa2_prefill_fig = get_fa2_prefill_fig(filtered_df)
|
117 |
filtered_fa2_decode_fig = get_fa2_decode_fig(filtered_df)
|
118 |
+
filtered_exllama_prefill_fig = get_exllama_prefill_fig(filtered_df)
|
119 |
+
filtered_exllama_decode_fig = get_exllama_decode_fig(filtered_df)
|
120 |
|
121 |
return [
|
122 |
filtered_leaderboard_df,
|
|
|
125 |
filtered_bt_decode_fig,
|
126 |
filtered_fa2_prefill_fig,
|
127 |
filtered_fa2_decode_fig,
|
128 |
+
filtered_exllama_prefill_fig,
|
129 |
+
filtered_exllama_decode_fig,
|
130 |
]
|
131 |
|
132 |
|