codingma
2024-04-07 11:34:01 +08:00
parent a88fe8c1af
commit 2565a32bd9
2 changed files with 23 additions and 0 deletions

View File

@@ -773,3 +773,12 @@ _register_template(
format_user=StringFormatter(slots=["<human>:{{content}}\n<bot>:"]),
format_separator=EmptyFormatter(slots=["\n"]),
)
_register_template(
name="mediatek",
format_user=StringFormatter(slots=["[INST] {{content}} [/INST] "]),
format_system=StringFormatter(slots=[{"bos_token"}, "{{content}}"]),
default_system="You are a helpful AI assistant built by MediaTek Research. The user you are helping speaks Traditional Chinese and comes from Taiwan. ",
force_system=True,
efficient_eos=True,
)