[model] add smollm2 support (#8220)

This commit is contained in:
Akshat Sehgal
2025-05-31 01:29:01 -07:00
committed by GitHub
parent 3e1a7fcb9c
commit c7e63bead7
2 changed files with 24 additions and 1 deletions

View File

@@ -1693,6 +1693,22 @@ register_template(
)
register_template(
name="smollm2",
format_system=StringFormatter(
slots=["<|im_start|>system\n{{content}}<|im_end|>\n"]
),
format_user=StringFormatter(
slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]
),
format_assistant=StringFormatter(
slots=["{{content}}<|im_end|>\n"]
),
stop_words=["<|im_end|>"],
default_system="You are a helpful AI assistant named SmolLM, trained by Hugging Face.",
)
register_template(
name="solar",
format_user=StringFormatter(slots=["### User:\n{{content}}\n\n### Assistant:\n"]),