wenjiao commited on
Commit
22b8709
1 Parent(s): 0567fb7

update Double Quant value

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -305,7 +305,11 @@ def update_table(
305
  except ValueError:
306
  group_dtype = [-1]
307
 
308
- double_quant = [str_to_bool(double_quant)]
 
 
 
 
309
  filtered_df = filter_models(df=hidden_df, type_query=type_query, size_query=size_query, precision_query=precision_query, hide_models=hide_models, compute_dtype=compute_dtype, weight_dtype=weight_dtype, double_quant=double_quant, group_dtype=group_dtype, params_query=params_query)
310
  filtered_df = filter_queries(query, filtered_df)
311
  df = select_columns(filtered_df, columns)
@@ -536,7 +540,7 @@ with demo:
536
  with gr.Row():
537
  filter_columns_computeDtype = gr.Dropdown(choices=[i.value.name for i in ComputeDtype], label="Compute Dtype", multiselect=False, value="All", interactive=True,)
538
  filter_columns_weightDtype = gr.Dropdown(choices=[i.value.name for i in WeightDtype], label="Weight Dtype", multiselect=False, value="All", interactive=True,)
539
- filter_columns_doubleQuant = gr.Dropdown(choices=["True", "False"], label="Double Quant", multiselect=False, value="False", interactive=True)
540
  filter_columns_groupDtype = gr.Dropdown(choices=[i.value.name for i in GroupDtype], label="Group Size", multiselect=False, value="All", interactive=True,)
541
 
542
  with gr.Row():
 
305
  except ValueError:
306
  group_dtype = [-1]
307
 
308
+ if double_quant == 'All':
309
+ double_quant = [True, False]
310
+ else:
311
+ double_quant = [str_to_bool(double_quant)]
312
+
313
  filtered_df = filter_models(df=hidden_df, type_query=type_query, size_query=size_query, precision_query=precision_query, hide_models=hide_models, compute_dtype=compute_dtype, weight_dtype=weight_dtype, double_quant=double_quant, group_dtype=group_dtype, params_query=params_query)
314
  filtered_df = filter_queries(query, filtered_df)
315
  df = select_columns(filtered_df, columns)
 
540
  with gr.Row():
541
  filter_columns_computeDtype = gr.Dropdown(choices=[i.value.name for i in ComputeDtype], label="Compute Dtype", multiselect=False, value="All", interactive=True,)
542
  filter_columns_weightDtype = gr.Dropdown(choices=[i.value.name for i in WeightDtype], label="Weight Dtype", multiselect=False, value="All", interactive=True,)
543
+ filter_columns_doubleQuant = gr.Dropdown(choices=["All", "True", "False"], label="Double Quant", multiselect=False, value="All", interactive=True)
544
  filter_columns_groupDtype = gr.Dropdown(choices=[i.value.name for i in GroupDtype], label="Group Size", multiselect=False, value="All", interactive=True,)
545
 
546
  with gr.Row():