wehub-resource-sync
25 lines
780 B
Python
25 lines
780 B
Python
"""RWKV default templates"""
|
|
|
|
from mlc_llm.protocol.conversation_protocol import Conversation, MessagePlaceholders
|
|
|
|
from .registry import ConvTemplateRegistry
|
|
|
|
# RWKV World
|
|
ConvTemplateRegistry.register_conv_template(
|
|
Conversation(
|
|
name="rwkv_world",
|
|
system_template=f"User: hi\n\nAssistant: {MessagePlaceholders.SYSTEM.value}",
|
|
system_message=(
|
|
"Hi. I am your assistant and I will provide expert full response "
|
|
"in full details. Please feel free to ask any question and I will "
|
|
"always answer it."
|
|
),
|
|
roles={"user": "User", "assistant": "Assistant"},
|
|
seps=["\n\n"],
|
|
role_content_sep=": ",
|
|
role_empty_sep=": ",
|
|
stop_str=["\n\n"],
|
|
stop_token_ids=[0],
|
|
)
|
|
)
|