Skip to content

Commit 0232cf9

Browse files
authored
[bugfix] fix from_dict (#5506)
1 parent ddd7f3c commit 0232cf9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

swift/trainers/rlhf_trainer/grpo_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1229,7 +1229,7 @@ def _apply_chat_template_to_messages_list(self, messages_list: InputsType):
12291229
prompts_text = []
12301230
for messages in messages_list:
12311231
InferRequest.remove_response(messages)
1232-
template_inputs, _ = StdTemplateInputs.from_dict({'messages': messages})
1232+
template_inputs = StdTemplateInputs.from_dict({'messages': messages})
12331233
res = self.template.encode(template_inputs)
12341234
prompts_text.append(self.template.safe_decode(res['input_ids']))
12351235
return prompts_text

0 commit comments

Comments
 (0)