fix: bugs

2025-12-17 20:30:36 +08:00 · 2024-12-19 21:08:16 +08:00
parent 7eb49e5ffa
commit d0eb64d5e3
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