From 11b55b7ce3f0c71300a17ead45da7a977388282b Mon Sep 17 00:00:00 2001 From: CJack812 <1070767757@qq.com> Date: Wed, 15 Oct 2025 23:52:04 +0800 Subject: [PATCH] Update loss_scale method call to pass through inputs.extra_kwargs --- swift/llm/template/base.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/swift/llm/template/base.py b/swift/llm/template/base.py index 238a4a89af..104f59e00b 100644 --- a/swift/llm/template/base.py +++ b/swift/llm/template/base.py @@ -1146,7 +1146,8 @@ def _swift_encode(self, inputs: StdTemplateInputs): if template_meta.auto_add_bos and sep_token: res_context_list.append(sep_token) res_context_types.append(ContextType.SUFFIX) - res_context_list, loss_scale_list = self.loss_scale(res_context_list, res_context_types, inputs.messages) + res_context_list, loss_scale_list = self.loss_scale(res_context_list, res_context_types, inputs.messages, + **inputs.extra_kwargs) if self.is_training: answer_len = len(extra_context_list) + bool(response is not None) else: