add pixtral template

This commit is contained in:
Kingsley
2024-09-26 12:11:58 +08:00
parent af8dfb58fd
commit 86f5a9be54
4 changed files with 61 additions and 0 deletions

View File

@@ -821,6 +821,13 @@ _register_template(
replace_eos=True,
)
_register_template(
name="pixtral",
format_user=StringFormatter(slots=["[INST] {{content}} [/INST]"]),
format_prefix=EmptyFormatter(slots=[{"bos_token"}]),
mm_plugin=get_mm_plugin(name="pixtral", image_token="[IMG]")
)
_register_template(
name="qwen",