Skip to content

Commit 588c25c

Browse files
committed
[bugfix] fix template extra_kwargs (#5821)
1 parent 04f66b7 commit 588c25c

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

swift/llm/template/template_inputs.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -269,13 +269,7 @@ def __post_init__(self):
269269
if not isinstance(value_dict, dict):
270270
continue
271271
if key in {'chosen', 'rejected'}:
272-
kwargs = {}
273-
for k in all_keys:
274-
val = value_dict.get(k)
275-
if val is None:
276-
continue
277-
kwargs[k] = val
278-
setattr(self, key, StdTemplateInputs.from_dict(kwargs))
272+
setattr(self, key, StdTemplateInputs.from_dict(value_dict))
279273
else:
280274
res = []
281275
for i in range(len(value_dict['messages'])):

0 commit comments

Comments
 (0)