ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: optimize frontend

Browse Source 
Former-commit-id: 4a78603c141d9bd78bcaf81261b443cf082bf51f
This commit is contained in:
ZeYi Lin
2024-12-19 19:04:19 +08:00
parent 44dfbf9dbd
commit 44895ebe36
4 changed files with 126 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/train/trainer_utils.py
View File

@@ -465,7 +465,7 @@ def get_swanlab_callback(finetuning_args: "FinetuningArguments") -> "TrainerCall
"""
import swanlab
from swanlab.integration.transformers import SwanLabCallback
if finetuning_args.swanlab_api_key is not None:
swanlab.login(api_key=finetuning_args.swanlab_api_key)