wenjiao commited on
Commit
d981095
1 Parent(s): 08847a4

modify update_table logic

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -117,7 +117,12 @@ def update_table(
117
  ):
118
 
119
  compute_dtype = [compute_dtype]
120
- weight_dtype = [weight_dtype]
 
 
 
 
 
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="Precision",
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,