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

[webui] upgrade to gradio 5 (#6688)

Browse Source 
Former-commit-id: 9df7721264ddef0008d7648e6ed173adef99bd74
This commit is contained in:
hoshi-hiyouga
2025-01-17 20:15:42 +08:00
committed by GitHub
parent 33525a34b6
commit 31daa6570b
8 changed files with 36 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -144,8 +144,7 @@ class Runner:
mask_history=get("train.mask_history"),
resize_vocab=get("train.resize_vocab"),
use_llama_pro=get("train.use_llama_pro"),
shift_attn=get("train.shift_attn"),
report_to="all" if get("train.report_to") else "none",
report_to=get("train.report_to"),
use_galore=get("train.use_galore"),
use_apollo=get("train.use_apollo"),
use_badam=get("train.use_badam"),
@@ -239,6 +238,12 @@ class Runner:
args["badam_switch_interval"] = get("train.badam_switch_interval")
args["badam_update_ratio"] = get("train.badam_update_ratio")
# report_to
if "none" in args["report_to"]:
args["report_to"] = "none"
elif "all" in args["report_to"]:
args["report_to"] = "all"
# swanlab config
if get("train.use_swanlab"):
args["swanlab_project"] = get("train.swanlab_project")