From b512a06c3d2aac2391f7f5a8988facda9c7e7452 Mon Sep 17 00:00:00 2001 From: ZeYi Lin <944270057@qq.com> Date: Thu, 19 Dec 2024 20:22:36 +0800 Subject: [PATCH] docs: config framework Former-commit-id: 7eb49e5ffaea59d8a2756ae7ff55bd57b9077f4b --- src/llamafactory/hparams/finetuning_args.py | 4 ++++ 1 file changed, 4 insertions(+) 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