feat: ui improve

Former-commit-id: 5f6dafd70e962b8fe9a294d555133002135f80df

src/llamafactory/hparams/finetuning_args.py

@@ -322,11 +322,7 @@ class SwanLabArguments:
         default=None,
         metadata={"help": "The experiment name in SwanLab."},
     )
-    swanlab_description: str = field(
+    swanlab_mode: Literal["cloud", "local"] = field(
-        default=None,
-        metadata={"help": "The experiment description in SwanLab."},
-    )
-    swanlab_mode: Literal["cloud", "local", "disabled"] = field(
         default="cloud",
         metadata={"help": "The mode of SwanLab."},
     )

src/llamafactory/train/trainer_utils.py

@@ -473,7 +473,6 @@ def get_swanlab_callback(finetuning_args: "FinetuningArguments") -> "TrainerCall
         project=finetuning_args.swanlab_project,
         workspace=finetuning_args.swanlab_workspace,
         experiment_name=finetuning_args.swanlab_experiment_name,
-        description=finetuning_args.swanlab_description,
         mode=finetuning_args.swanlab_mode,
         config={"Framework": "🦙LLaMA Factory"},
     )

src/llamafactory/webui/components/train.py

@@ -275,12 +275,11 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
             use_swanlab = gr.Checkbox()
             swanlab_project = gr.Textbox(value="LLaMA-Factory", placeholder="Project name", interactive=True)
             swanlab_experiment_name = gr.Textbox(value="", placeholder="Experiment name", interactive=True)
-            swanlab_description = gr.Textbox(value="", placeholder="Experiment description", interactive=True)
-            swanlab_mode = gr.Dropdown(choices=["cloud", "local", "disabled"], value="cloud", interactive=True)
             swanlab_workspace = gr.Textbox(value="", placeholder="Workspace name", interactive=True)
             swanlab_api_key = gr.Textbox(value="", placeholder="API key", interactive=True)
+            swanlab_mode = gr.Dropdown(choices=["cloud", "local", "disabled"], value="cloud", interactive=True)
 
-    input_elems.update({use_swanlab, swanlab_api_key, swanlab_project, swanlab_workspace, swanlab_experiment_name, swanlab_description, swanlab_mode})
+    input_elems.update({use_swanlab, swanlab_api_key, swanlab_project, swanlab_workspace, swanlab_experiment_name, swanlab_mode})
     elem_dict.update(
         dict(
             swanlab_tab=swanlab_tab,
@@ -289,7 +288,6 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
             swanlab_project=swanlab_project,
             swanlab_workspace=swanlab_workspace,
             swanlab_experiment_name=swanlab_experiment_name,
-            swanlab_description=swanlab_description,
             swanlab_mode=swanlab_mode,
         )
     )

src/llamafactory/webui/locales.py

@@ -1387,90 +1387,85 @@ LOCALES = {
     },
     "swanlab_api_key": {
         "en": {
-            "label": "API key",
+            "label": "API Key(optional)",
-            "info": "The API key for SwanLab.",
+            "info": "API key for SwanLab. Once logged in, no need to login again in the programming environment.",
         },
         "ru": {
-            "label": "API ключ",
+            "label": "API ключ(Необязательный)",
-            "info": "API ключ для SwanLab.",
+            "info": "API ключ для SwanLab. После входа в программное окружение, нет необходимости входить снова.",
         },
         "zh": {
-            "label": "API 密钥",
+            "label": "API密钥(选填)",
-            "info": "SwanLab 的 API 密钥。",
+            "info": "用于在编程环境登录SwanLab，已登录则无需填写。",
         },
         "ko": {
-            "label": "API 키",
+            "label": "API 키(선택 사항)",
-            "info": "SwanLab의 API 키.",
+            "info": "SwanLab의 API 키. 프로그래밍 환경에 로그인한 후 다시 로그인할 필요가 없습니다.",
         },
     },
     "swanlab_project": {
         "en": {
-            "label": "SwanLab project",
+            "label": "Project(optional)",
         },
         "ru": {
-            "label": "Проект SwanLab",
+            "label": "Проект(Необязательный)",
         },
         "zh": {
-            "label": "SwanLab 项目",
+            "label": "项目(选填)",
         },
         "ko": {
-            "label": "SwanLab 프로젝트",
+            "label": "프로젝트(선택 사항)",
         },
     },
     "swanlab_workspace": {
         "en": {
-            "label": "SwanLab workspace",
+            "label": "Workspace(optional)",
+            "info": "Workspace for SwanLab. If not filled, it defaults to the personal workspace.",
+            
         },
         "ru": {
-            "label": "Рабочая область SwanLab",
+            "label": "Рабочая область(Необязательный)",
+            "info": "Рабочая область SwanLab, если не заполнено, то по умолчанию в личной рабочей области.",
         },
         "zh": {
-            "label": "SwanLab 工作区",
+            "label": "Workspace(选填)",
+            "info": "SwanLab组织的工作区，如不填写则默认在个人工作区下",
         },
         "ko": {
-            "label": "SwanLab 작업 영역",
+            "label": "작업 영역(선택 사항)",
+            "info": "SwanLab 조직의 작업 영역, 비어 있으면 기본적으로 개인 작업 영역에 있습니다.",
         },
     },
     "swanlab_experiment_name": {
         "en": {
-            "label": "SwanLab experiment name",
+            "label": "Experiment_name(optional)",
         },
         "ru": {
-            "label": "Имя эксперимента SwanLab",
+            "label": "Имя эксперимента(Необязательный)",
         },
         "zh": {
-            "label": "SwanLab 实验名称",
+            "label": "实验名(选填)  ",
         },
         "ko": {
-            "label": "SwanLab 실험 이름",
+            "label": "실험 이름(선택 사항)",
-        },
-    },
-    "swanlab_description": {
-        "en": {
-            "label": "SwanLab experiment description",
-        },
-        "ru": {
-            "label": "Описание эксперимента SwanLab",
-        },
-        "zh": {
-            "label": "SwanLab 实验描述",
-        },
-        "ko": {
-            "label": "SwanLab 실험 설명",
         },
     },
     "swanlab_mode": {
         "en": {
-            "label": "SwanLab mode",
+            "label": "Mode",
+            "info": "Cloud or offline version.",    
         },
         "ru": {
-            "label": "Режим SwanLab",
+            "label": "Режим",
+            "info": "Версия в облаке или локальная версия.",
         },
         "zh": {
-            "label": "SwanLab 模式",
+            "label": "模式",
+            "info": "云端版或离线版",
         },
         "ko": {
-            "label": "SwanLab 모드",
+            "label": "모드",
+            "info": "클라우드 버전 또는 오프라인 버전.",
         },
     },
     "cmd_preview_btn": {

src/llamafactory/webui/runner.py

@@ -235,7 +235,6 @@ class Runner:
             args["swanlab_project"] = get("train.swanlab_project")
             args["swanlab_workspace"] = get("train.swanlab_workspace")
             args["swanlab_experiment_name"] = get("train.swanlab_experiment_name")
-            args["swanlab_description"] = get("train.swanlab_description")
             args["swanlab_mode"] = get("train.swanlab_mode")