ishaansehgal99 commited on
Commit
f941d7f
·
verified ·
1 Parent(s): ef38235

feat: Add chat template

Browse files

got it from https://huggingface.co/microsoft/phi-4/blob/main/tokenizer_config.json

Files changed (1) hide show
  1. tokenizer_config.json +2 -1
tokenizer_config.json CHANGED
@@ -319,5 +319,6 @@
319
  "eos_token": "<|endoftext|>",
320
  "model_max_length": 2048,
321
  "tokenizer_class": "CodeGenTokenizer",
322
- "unk_token": "<|endoftext|>"
 
323
  }
 
319
  "eos_token": "<|endoftext|>",
320
  "model_max_length": 2048,
321
  "tokenizer_class": "CodeGenTokenizer",
322
+ "unk_token": "<|endoftext|>",
323
+ "chat_template": "{% for message in messages %}{% if (message['role'] == 'system') %}{{'<|im_start|>system<|im_sep|>' + message['content'] + '<|im_end|>'}}{% elif (message['role'] == 'user') %}{{'<|im_start|>user<|im_sep|>' + message['content'] + '<|im_end|>'}}{% elif (message['role'] == 'assistant') %}{{'<|im_start|>assistant<|im_sep|>' + message['content'] + '<|im_end|>'}}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant<|im_sep|>' }}{% endif %}"
324
  }