fix bug (#267)

Author: tastelikefeet

swift/ui/base.py

@@ -31,7 +31,7 @@ def wrapper(*args, **kwargs):
         if 'is_list' in kwargs:
             self.is_list = kwargs.pop('is_list')
 
-        if base_builder and base_builder.default(elem_id):
+        if base_builder and base_builder.default(elem_id) is not None:
             kwargs['value'] = base_builder.default(elem_id)
 
         if builder is not None:
@@ -107,7 +107,7 @@ def default(cls, elem_id):
             _choice = sub_ui.default(elem_id)
             if _choice:
                 return _choice
-        return cls.default_dict.get(elem_id, [])
+        return cls.default_dict.get(elem_id, None)
 
     @classmethod
     def locale(cls, elem_id, lang):

swift/ui/llm_train/quantization.py

@@ -50,8 +50,7 @@ class Quantization(BaseUI):
     def do_build_ui(cls, base_tab: Type['BaseUI']):
         with gr.Accordion(elem_id='quantization_tab', open=False):
             with gr.Row():
-                gr.Slider(
-                    elem_id='quantization_bit', minimum=0, maximum=8, step=4)
+                gr.Dropdown(elem_id='quantization_bit')
                 gr.Dropdown(elem_id='bnb_4bit_comp_dtype')
                 gr.Dropdown(elem_id='bnb_4bit_quant_type')
                 gr.Checkbox(elem_id='bnb_4bit_use_double_quant')