423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 03:40:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

update ui

Browse Source
This commit is contained in:
hiyouga
2024-07-03 23:13:49 +08:00
parent a4a1ddbcb9
commit 7f770f6895
3 changed files with 23 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -138,10 +138,9 @@ class Runner:
warmup_steps=get("train.warmup_steps"),
neftune_noise_alpha=get("train.neftune_alpha") or None,
optim=get("train.optim"),
packing=get("train.packing") or get("train.neat_packing"),
neat_packing=get("train.neat_packing"),
resize_vocab=get("train.resize_vocab"),
packing=get("train.packing"),
efficient_packing=get("train.efficient_packing"),
upcast_layernorm=get("train.upcast_layernorm"),
use_llama_pro=get("train.use_llama_pro"),
shift_attn=get("train.shift_attn"),
report_to="all" if get("train.report_to") else "none",