diff --git a/src/llamafactory/hparams/finetuning_args.py b/src/llamafactory/hparams/finetuning_args.py index 72b6a0a9..b39e9f18 100644 --- a/src/llamafactory/hparams/finetuning_args.py +++ b/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( - default=None, - metadata={"help": "The experiment description in SwanLab."}, - ) - swanlab_mode: Literal["cloud", "local", "disabled"] = field( + swanlab_mode: Literal["cloud", "local"] = field( default="cloud", metadata={"help": "The mode of SwanLab."}, ) diff --git a/src/llamafactory/train/trainer_utils.py b/src/llamafactory/train/trainer_utils.py index 2a50bf71..5b8fb403 100644 --- a/src/llamafactory/train/trainer_utils.py +++ b/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"}, ) diff --git a/src/llamafactory/webui/components/train.py b/src/llamafactory/webui/components/train.py index d7237bd6..8766d5b9 100644 --- a/src/llamafactory/webui/components/train.py +++ b/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, ) ) diff --git a/src/llamafactory/webui/locales.py b/src/llamafactory/webui/locales.py index 2167bfd1..e267b63c 100644 --- a/src/llamafactory/webui/locales.py +++ b/src/llamafactory/webui/locales.py @@ -1387,90 +1387,85 @@ LOCALES = { }, "swanlab_api_key": { "en": { - "label": "API key", - "info": "The API key for SwanLab.", + "label": "API Key(optional)", + "info": "API key for SwanLab. Once logged in, no need to login again in the programming environment.", }, "ru": { - "label": "API ключ", - "info": "API ключ для SwanLab.", + "label": "API ключ(Необязательный)", + "info": "API ключ для SwanLab. После входа в программное окружение, нет необходимости входить снова.", }, "zh": { - "label": "API 密钥", - "info": "SwanLab 的 API 密钥。", + "label": "API密钥(选填)", + "info": "用于在编程环境登录SwanLab，已登录则无需填写。", }, "ko": { - "label": "API 키", - "info": "SwanLab의 API 키.", + "label": "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 실험 이름", - }, - }, - "swanlab_description": { - "en": { - "label": "SwanLab experiment description", - }, - "ru": { - "label": "Описание эксперимента SwanLab", - }, - "zh": { - "label": "SwanLab 实验描述", - }, - "ko": { - "label": "SwanLab 실험 설명", + "label": "실험 이름(선택 사항)", }, }, "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": { diff --git a/src/llamafactory/webui/runner.py b/src/llamafactory/webui/runner.py index 6c2b4feb..2b5c55b8 100644 --- a/src/llamafactory/webui/runner.py +++ b/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")