Skip to content

Commit 28f77bf

Browse files
authored
[grpo] fix apply_chat_template (#4827)
1 parent a61c319 commit 28f77bf

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

swift/trainers/rlhf_trainer/grpo_trainer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1094,7 +1094,8 @@ def _apply_chat_template_to_messages_list(self, messages_list: InputsType):
10941094
InferRequest.remove_response(messages)
10951095
template_inputs, _ = StdTemplateInputs.from_dict({'messages': messages})
10961096
res_context_list, _, _ = self.template._swift_encode(template_inputs)
1097-
prompts_text.append(''.join(res_context_list))
1097+
prompts_text.append(''.join(elem for elem in res_context_list if isinstance(elem, str)))
1098+
10981099
return prompts_text
10991100

11001101
@profiling_decorator

0 commit comments

Comments
 (0)