[3rdparty] support swanlab lark notification (#7481)

2025-03-27 01:52:01 +08:00
parent 24afceddb7
commit b578a7d5b6
2 changed files with 17 additions and 0 deletions
--- a/src/llamafactory/hparams/finetuning_args.py
+++ b/src/llamafactory/hparams/finetuning_args.py
@@ -375,6 +375,14 @@ class SwanLabArguments:
        default=None,
        metadata={"help": "The log directory for SwanLab."},
    )
+    swanlab_lark_webhook_url: Optional[str] = field(
+        default=None,
+        metadata={"help": "The Lark(飞书) webhook URL for SwanLab."},
+    )
+    swanlab_lark_secret: Optional[str] = field(
+        default=None,
+        metadata={"help": "The Lark(飞书) secret for SwanLab."},
+    )


@dataclass