Skip to content

Commit dafdf2f

Browse files
committed
cleanup conv template
1 parent a178922 commit dafdf2f

File tree

1 file changed

+3
-7
lines changed
  • python/mlc_llm/conversation_template

1 file changed

+3
-7
lines changed

python/mlc_llm/conversation_template/llama.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,6 @@
1111
Conversation(
1212
name="llama-4",
1313
system_template="",
14-
# system_template=(
15-
# # "<|header_start|>user<|header_end|>\n\n" #system
16-
# # f"{MessagePlaceholders.SYSTEM.value}" #<|eot|>
17-
# ),
1814
system_message="",
1915
roles={
2016
"user": "<|header_start|>user",
@@ -24,9 +20,9 @@
2420
seps=["<|eot|>"],
2521
role_content_sep="<|header_end|>\n\n",
2622
role_empty_sep="<|header_end|>\n\n",
27-
stop_str=[], #"<|end_of_text|>", "<|eom|>", "<|eot|>"
28-
stop_token_ids=[200001, 200007, 200008],#[128001, 128008, 128009], # "<|end_of_text|>", "<|eom|>", "<|eot|>"
29-
system_prefix_token_ids=[200000],#[128000], # "<|begin_of_text|>"
23+
stop_str=[],
24+
stop_token_ids=[200001, 200007, 200008],# "<|end_of_text|>", "<|eom|>", "<|eot|>"
25+
system_prefix_token_ids=[200000],# "<|begin_of_text|>"
3026
add_role_after_system_message=False,
3127
)
3228
)

0 commit comments

Comments
 (0)