feat: Add chat template
#137
by ishaansehgal99 - opened
- 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 |
}
|