From d7e35ddcaabb1edd7976a8e28f30bc6937ded5ed 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: eb294fed1319e97657b54b9aa479d87734c711dc
---
 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"):