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

Merge pull request #5971 from hiyouga/hiyouga/fix_webui

Browse Source 
[webui] fix extra args
This commit is contained in:
hoshi-hiyouga
2024-11-09 00:25:24 +08:00
committed by GitHub
parent 39e330196d eb294fed13
commit adc5849ce7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -154,8 +154,8 @@ class Runner:
plot_loss=True, plot_loss=True,
ddp_timeout=180000000, ddp_timeout=180000000,
include_num_input_tokens_seen=False if is_transformers_version_equal_to_4_46() else True, # FIXME include_num_input_tokens_seen=False if is_transformers_version_equal_to_4_46() else True, # FIXME
**json.loads(get("train.extra_args")),
) )
args.update(json.loads(get("train.extra_args")))
# checkpoints # checkpoints
if get("top.checkpoint_path"): if get("top.checkpoint_path"):