diff --git a/src/llamafactory/train/trainer_utils.py b/src/llamafactory/train/trainer_utils.py index e626c0f0..87396220 100644 --- 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