ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[model] add smollm2 and medgemma (#8161)

Browse Source
This commit is contained in:
hoshi-hiyouga
2025-05-26 23:19:58 +08:00
committed by GitHub
parent ba032828e2
commit 3c7dc66a92
4 changed files with 54 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/llamafactory/data/template.py
View File

@@ -1686,15 +1686,9 @@ register_template(
register_template(
name="smollm",
format_system=StringFormatter(
slots=["<|im_start|>system\n{{content}}<|im_end|>\n"]
),
format_user=StringFormatter(
slots=["<|im_start|>user\n{{content}}<|im_end|>\n"]
),
format_assistant=StringFormatter(
slots=["<|im_start|>assistant\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"]),
format_system=StringFormatter(slots=["<|im_start|>system\n{{content}}<|im_end|>\n"]),
stop_words=["<|im_end|>"],
)