Skip to content

Commit 709f484

Browse files
authored
[ConvTemplate] Update Gemma template with <bos> (#2722)
This commit adds `<bos>` to the gemma's conversation template.
1 parent b0f2731 commit 709f484

File tree

1 file changed

+1
-1
lines changed
  • python/mlc_llm/conversation_template

1 file changed

+1
-1
lines changed

python/mlc_llm/conversation_template/gemma.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
ConvTemplateRegistry.register_conv_template(
99
Conversation(
1010
name="gemma_instruction",
11-
system_template=f"{MessagePlaceholders.SYSTEM.value}",
11+
system_template=f"<bos>{MessagePlaceholders.SYSTEM.value}",
1212
system_message="",
1313
roles={"user": "<start_of_turn>user", "assistant": "<start_of_turn>model"},
1414
seps=["<end_of_turn>\n"],

0 commit comments

Comments
 (0)