codingma committed 2024-04-07 11:34:01 +08:00
1 parent a88fe8c1af
commit 2565a32bd9
2 files changed
+23

No files matched your search

+9
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,
)