diff --git a/src/llamafactory/webui/runner.py b/src/llamafactory/webui/runner.py index c50d7759..3dd80f6b 100644 --- a/src/llamafactory/webui/runner.py +++ b/src/llamafactory/webui/runner.py @@ -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"):