modify update_table logic
Browse files
app.py
CHANGED
@@ -117,7 +117,12 @@ def update_table(
|
|
117 |
):
|
118 |
|
119 |
compute_dtype = [compute_dtype]
|
120 |
-
|
|
|
|
|
|
|
|
|
|
|
121 |
if group_dtype == 'All':
|
122 |
group_dtype = [-1, 1024, 256, 128, 64, 32]
|
123 |
else:
|
@@ -282,7 +287,7 @@ with demo:
|
|
282 |
elem_id="filter-columns-type",
|
283 |
)
|
284 |
filter_columns_precision = gr.CheckboxGroup(
|
285 |
-
label="
|
286 |
choices=[i.value.name for i in Precision],
|
287 |
value=[i.value.name for i in Precision],
|
288 |
interactive=True,
|
|
|
117 |
):
|
118 |
|
119 |
compute_dtype = [compute_dtype]
|
120 |
+
|
121 |
+
if weight_dtype == 'All':
|
122 |
+
weight_dtype = ['int2', 'int3', 'int4', 'nf4', 'fp4']
|
123 |
+
else:
|
124 |
+
weight_dtype = [weight_dtype]
|
125 |
+
|
126 |
if group_dtype == 'All':
|
127 |
group_dtype = [-1, 1024, 256, 128, 64, 32]
|
128 |
else:
|
|
|
287 |
elem_id="filter-columns-type",
|
288 |
)
|
289 |
filter_columns_precision = gr.CheckboxGroup(
|
290 |
+
label="Weight precision",
|
291 |
choices=[i.value.name for i in Precision],
|
292 |
value=[i.value.name for i in Precision],
|
293 |
interactive=True,
|