IlyasMoutawwakil HF staff commited on
Commit
affd732
ยท
1 Parent(s): 2460b35

fix filters

Browse files
Files changed (1) hide show
  1. 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