Fix chatml template (#406)

This commit is contained in:
Lianmin Zheng
2024-04-30 15:53:39 -07:00
committed by GitHub
parent 95c4e0dfac
commit ba67101f99

View File

@@ -400,7 +400,7 @@ register_conv_template(
Conversation(
name="chatml",
system_template="<|im_start|>system\n{system_message}",
system_message="You are an AI assistant.",
system_message="You are a helpful assistant.",
roles=("<|im_start|>user", "<|im_start|>assistant"),
sep_style=SeparatorStyle.CHATML,
sep="<|im_end|>",