fix: bugs

Former-commit-id: a2297f97f7587c77d55fbce9ffa81dc60d0b04a1
2024-12-19 21:08:16 +08:00
parent a30bdbb1c0
commit 925e421bde
3 changed files with 2 additions and 5 deletions
--- a/src/llamafactory/train/trainer_utils.py
+++ b/src/llamafactory/train/trainer_utils.py
@@ -475,6 +475,7 @@ def get_swanlab_callback(finetuning_args: "FinetuningArguments") -> "TrainerCall
        experiment_name=finetuning_args.swanlab_experiment_name,
        description=finetuning_args.swanlab_description,
        mode=finetuning_args.swanlab_mode,
+        config={"Framework": "🦙LLaMA Factory"},
    )

    return swanlab_callback