add chat template (#1)
Browse files- add chat template (48984a7e8626211580e20dcf3cc22f5cd3147f3f)
Co-authored-by: Loubna Ben Allal <[email protected]>
- tokenizer_config.json +1 -0
tokenizer_config.json
CHANGED
@@ -303,6 +303,7 @@
|
|
303 |
"<pr_diff_hunk_comment_line>"
|
304 |
],
|
305 |
"bos_token": "<|endoftext|>",
|
|
|
306 |
"clean_up_tokenization_spaces": true,
|
307 |
"eos_token": "<|endoftext|>",
|
308 |
"model_max_length": 1000000000000000019884624838656,
|
|
|
303 |
"<pr_diff_hunk_comment_line>"
|
304 |
],
|
305 |
"bos_token": "<|endoftext|>",
|
306 |
+
"chat_template": "{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
|
307 |
"clean_up_tokenization_spaces": true,
|
308 |
"eos_token": "<|endoftext|>",
|
309 |
"model_max_length": 1000000000000000019884624838656,
|