From 0027f46cccee430afa21e9a7cc98b9c94b1f0ac8 Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Sat, 9 Nov 2024 00:24:27 +0800
Subject: [PATCH] fix extra args

Former-commit-id: 2c98a1bc3d885170f8298872c2ea2e24427fb447
---
 src/llamafactory/webui/runner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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"):