If your model expects special tokens, make sure to add them to the template!