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

8
src/llamafactory/webui/runner.py
View File

@@ -231,7 +231,13 @@ class Runner:
# swanlab config
if get("train.use_swanlab"):
args["swanlab_name"] = get("train.swanlab_name")
args["swanlab_api_key"] = get("train.swanlab_api_key")
args["swanlab_project"] = get("train.swanlab_project")
args["swanlab_workspace"] = get("train.swanlab_workspace")
args["swanlab_experiment_name"] = get("train.swanlab_experiment_name")
args["swanlab_description"] = get("train.swanlab_description")
args["swanlab_mode"] = get("train.swanlab_mode")
# eval config
if get("train.val_size") > 1e-6 and args["stage"] != "ppo":