update ComputeDtype logic

app.py

@@ -116,12 +116,15 @@ def update_table(
     group_dtype: str
 ):
 
-    compute_dtype = [compute_dtype]
-    
     if weight_dtype == 'All':
         weight_dtype = ['int2', 'int3', 'int4', 'nf4', 'fp4']
     else:
         weight_dtype = [weight_dtype]
+
+    if compute_dtype == 'All':
+        compute_dtype = ['bfloat16', 'float16', 'int8', 'float32']
+    else:
+        compute_dtype = [compute_dtype]   
         
     if group_dtype == 'All':
         group_dtype = [-1, 1024, 256, 128, 64, 32]
@@ -296,7 +299,7 @@ with demo:
                     with gr.Box() as config:
                         gr.HTML("""<p style='padding-bottom: 0.5rem; color: #6b7280; '>Quantization config</p>""")
                         with gr.Row():
-                            filter_columns_computeDtype = gr.Dropdown(choices=[i.value.name for i in ComputeDtype], label="Compute Dtype", multiselect=False, value="float16", interactive=True,)
+                            filter_columns_computeDtype = gr.Dropdown(choices=[i.value.name for i in ComputeDtype], label="Compute Dtype", multiselect=False, value="All", interactive=True,)
                             filter_columns_weightDtype = gr.Dropdown(choices=[i.value.name for i in WeightDtype], label="Weight Dtype", multiselect=False, value="All", interactive=True,)
                             filter_columns_doubleQuant = gr.Dropdown(choices=["True", "False"], label="Double Quant", multiselect=False, value=False, interactive=True)
                             filter_columns_groupDtype = gr.Dropdown(choices=[i.value.name for i in GroupDtype], label="Group Size", multiselect=False, value="All", interactive=True,)