wehub-resource-sync
26 lines
834 B
Python
26 lines
834 B
Python
"""LLM-jp default templates"""
|
|
|
|
from mlc_llm.protocol.conversation_protocol import Conversation, MessagePlaceholders
|
|
|
|
from .registry import ConvTemplateRegistry
|
|
|
|
# LLM-jp instruct
|
|
ConvTemplateRegistry.register_conv_template(
|
|
Conversation(
|
|
name="llm-jp",
|
|
system_template=f"{MessagePlaceholders.SYSTEM.value}",
|
|
system_message="以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。",
|
|
roles={
|
|
"user": "\n\n### 指示:",
|
|
"assistant": "\n\n### 応答:",
|
|
},
|
|
seps=["", "</s>"],
|
|
role_content_sep="\n",
|
|
role_empty_sep="\n",
|
|
stop_str=[],
|
|
stop_token_ids=[2], # eos_token_id
|
|
system_prefix_token_ids=[1], # bos_token_id (<s>)
|
|
add_role_after_system_message=True,
|
|
)
|
|
)
|