fix webui (#296)

Author: tastelikefeet

swift/ui/base.py

@@ -1,4 +1,6 @@
 import os
+import typing
+from dataclasses import fields
 from functools import partial, wraps
 from typing import Any, Dict, List, OrderedDict, Type
 
@@ -146,3 +148,23 @@ def set_lang(cls, lang):
         cls.lang = lang
         for sub_ui in cls.sub_ui:
             sub_ui.lang = lang
+
+    @staticmethod
+    def get_choices_from_dataclass(dataclass):
+        choice_dict = {}
+        for f in fields(dataclass):
+            if 'choices' in f.metadata:
+                choice_dict[f.name] = f.metadata['choices']
+            if 'Literal' in type(f.type).__name__ and typing.get_args(f.type):
+                choice_dict[f.name] = typing.get_args(f.type)
+        return choice_dict
+
+    @staticmethod
+    def get_default_value_from_dataclass(dataclass):
+        default_dict = {}
+        for f in fields(dataclass):
+            if hasattr(dataclass, f.name):
+                default_dict[f.name] = getattr(dataclass, f.name)
+            else:
+                default_dict[f.name] = None
+        return default_dict

swift/ui/llm_infer/llm_infer.py

@@ -78,10 +78,7 @@ class LLMInfer(BaseUI):
         },
     }
 
-    choice_dict = {}
-    for f in fields(InferArguments):
-        if 'choices' in f.metadata:
-            choice_dict[f.name] = f.metadata['choices']
+    choice_dict = BaseUI.get_choices_from_dataclass(InferArguments)
 
     @classmethod
     def do_build_ui(cls, base_tab: Type['BaseUI']):
@@ -143,11 +140,7 @@ def reset_memory(cls):
     def prepare_checkpoint(cls, *args):
         global model, tokenizer, template
         torch.cuda.empty_cache()
-        infer_args = fields(InferArguments)
-        infer_args = {
-            arg.name: getattr(InferArguments, arg.name)
-            for arg in infer_args
-        }
+        infer_args = cls.get_default_value_from_dataclass(InferArguments)
         kwargs = {}
         kwargs_is_list = {}
         other_kwargs = {}

swift/ui/llm_train/llm_train.py

@@ -1,7 +1,6 @@
 import os
 import sys
 import time
-from dataclasses import fields
 from typing import Dict, Type
 
 import gradio as gr
@@ -147,12 +146,8 @@ class LLMTrain(BaseUI):
         }
     }
 
-    choice_dict = {}
-    default_dict = {}
-    for f in fields(SftArguments):
-        if 'choices' in f.metadata:
-            choice_dict[f.name] = f.metadata['choices']
-        default_dict[f.name] = getattr(SftArguments, f.name)
+    choice_dict = BaseUI.get_choices_from_dataclass(SftArguments)
+    default_dict = BaseUI.get_default_value_from_dataclass(SftArguments)
 
     @classmethod
     def do_build_ui(cls, base_tab: Type['BaseUI']):
@@ -210,11 +205,7 @@ def do_build_ui(cls, base_tab: Type['BaseUI']):
     @classmethod
     def train(cls, *args):
         ignore_elements = ('model_type', 'logging_dir', 'more_params')
-        sft_args = fields(SftArguments)
-        sft_args = {
-            arg.name: getattr(SftArguments, arg.name)
-            for arg in sft_args
-        }
+        sft_args = cls.get_default_value_from_dataclass(SftArguments)
         kwargs = {}
         kwargs_is_list = {}
         other_kwargs = {}
@@ -244,7 +235,7 @@ def train(cls, *args):
         params = ''
 
         for e in kwargs:
-            if kwargs_is_list[e]:
+            if e in kwargs_is_list and kwargs_is_list[e]:
                 params += f'--{e} {kwargs[e]} '
             else:
                 params += f'--{e} "{kwargs[e]}" '