[tracking] swanlab add llamafactory tag (#8258)

2025-06-03 18:42:29 +08:00
parent 554e89ff02
commit c4e51d40e0
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