convert : fix smollm3 jinja template (#14586)

This commit is contained in:
Xuan-Son Nguyen
2025-07-09 08:26:13 +02:00
committed by GitHub
parent 6efcd65945
commit 20b7bf8a32

View File

@ -6692,6 +6692,16 @@ class HunYuanMoEModel(TextModel):
class SmolLM3Model(LlamaModel):
model_arch = gguf.MODEL_ARCH.SMOLLM3
def set_vocab(self):
super().set_vocab()
# remove unsupported array slicing in chat template
# ref: https://huggingface.co/ggml-org/SmolLM3-3B-GGUF/discussions/1
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained(self.dir_model)
if tokenizer.chat_template is not None:
chat_template = tokenizer.chat_template.replace("[:]", "")
self.gguf_writer.add_chat_template(chat_template)
###### CONVERSION LOGIC ######