diff --git a/src/llamafactory/hparams/finetuning_args.py b/src/llamafactory/hparams/finetuning_args.py index aaa28a6a..72b6a0a9 100644 --- a/src/llamafactory/hparams/finetuning_args.py +++ b/src/llamafactory/hparams/finetuning_args.py @@ -334,10 +334,6 @@ class SwanLabArguments: default=None, metadata={"help": "The API key for SwanLab."}, ) - swanlab_config: dict = field( - default={"Framework": "🦙LLaMA Factory"}, - metadata={"help": "The configuration file for SwanLab."}, - ) @dataclass diff --git a/src/llamafactory/train/trainer_utils.py b/src/llamafactory/train/trainer_utils.py index 583ca3ee..2a50bf71 100644 --- a/src/llamafactory/train/trainer_utils.py +++ b/src/llamafactory/train/trainer_utils.py @@ -475,6 +475,7 @@ def get_swanlab_callback(finetuning_args: "FinetuningArguments") -> "TrainerCall experiment_name=finetuning_args.swanlab_experiment_name, description=finetuning_args.swanlab_description, mode=finetuning_args.swanlab_mode, + config={"Framework": "🦙LLaMA Factory"}, ) return swanlab_callback \ No newline at end of file diff --git a/src/llamafactory/webui/components/train.py b/src/llamafactory/webui/components/train.py index f905e43e..6b14bb81 100644 --- a/src/llamafactory/webui/components/train.py +++ b/src/llamafactory/webui/components/train.py @@ -280,7 +280,7 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]: swanlab_description = gr.Textbox(value="", placeholder="Experiment description", interactive=True) swanlab_mode = gr.Dropdown(choices=["cloud", "local", "disabled"], value="cloud", interactive=True) - input_elems.update({use_swanlab}) + input_elems.update({use_swanlab, swanlab_api_key, swanlab_project, swanlab_workspace, swanlab_experiment_name, swanlab_description, swanlab_mode}) elem_dict.update( dict( swanlab_tab=swanlab_tab,