From dd6a9e4ede3fa02f8b0a01a52c5616792fa9ecb4 Mon Sep 17 00:00:00 2001
From: ViperEkura <3081035982@qq.com>
Date: Tue, 30 Sep 2025 17:30:00 +0800
Subject: [PATCH] =?UTF-8?q?fix(train.py):=20=E4=BF=AE=E5=A4=8D=E5=8F=82?=
 =?UTF-8?q?=E6=95=B0=E4=BC=A0=E9=80=92=E9=94=99=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 train.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index cc233e1..feb7f8b 100644
--- a/train.py
+++ b/train.py
@@ -52,12 +52,13 @@ def train(
         "dpo_beta": dpo_beta,
         "bos_token_id": parameter.tokenizer.bos_id,
         "eos_token_id": parameter.tokenizer.eos_id,
+        "pad_token_id": parameter.tokenizer.pad_id,
         "user_token_id":parameter.tokenizer.user_id,
     }
     
     strategy = StrategyFactory.load(
         model, 
-        train_type
+        train_type,
         **kwargs
     )
 
@@ -66,7 +67,7 @@ def train(
         load_path=cache_files,
         max_len=parameter.config.m_len,
         device=device,
-        dataset_kwargs=kwargs
+        **kwargs
     )
     
     param_groups = [