diff --git a/src/llamafactory/hparams/finetuning_args.py b/src/llamafactory/hparams/finetuning_args.py index 72b6a0a9..aaa28a6a 100644 --- a/src/llamafactory/hparams/finetuning_args.py +++ b/src/llamafactory/hparams/finetuning_args.py @@ -334,6 +334,10 @@ 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