[tracking] swanlab add llamafactory tag (#8258)

2025-08-02 03:32:50 +08:00 · 2025-06-03 18:42:29 +08:00 · 2025-06-03 18:42:29 +08:00 · 6cc247e815
commit 6cc247e815
parent 16a3f8a71b
1 changed files with 1 additions and 0 deletions
--- a/src/llamafactory/train/trainer_utils.py
+++ b/src/llamafactory/train/trainer_utils.py
@ -665,6 +665,7 @@ def get_swanlab_callback(finetuning_args: "FinetuningArguments") -> "TrainerCall
        mode=finetuning_args.swanlab_mode,
        config={"Framework": "🦙LlamaFactory"},
        logdir=finetuning_args.swanlab_logdir,
+        tags=["🦙LlamaFactory"],
    )
    return swanlab_callback