Merge pull request #5971 from hiyouga/hiyouga/fix_webui

[webui] fix extra args

Former-commit-id: adc5849ce75cc8eda800c73c59b1f44d3108e992
This commit is contained in:
hoshi-hiyouga 2024-11-09 00:25:24 +08:00 committed by GitHub
commit 1e00de38cb

View File

@ -154,8 +154,8 @@ class Runner:
plot_loss=True,
ddp_timeout=180000000,
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
if get("top.checkpoint_path"):