Daehyun Ahn
commited on
add chat_template to tokenizer
Browse files- tokenizer_config.json +1 -0
tokenizer_config.json
CHANGED
|
@@ -315,6 +315,7 @@
|
|
| 315 |
}
|
| 316 |
},
|
| 317 |
"bos_token": "<|endoftext|>",
|
|
|
|
| 318 |
"clean_up_tokenization_spaces": true,
|
| 319 |
"eos_token": "<|endoftext|>",
|
| 320 |
"model_max_length": 2048,
|
|
|
|
| 315 |
}
|
| 316 |
},
|
| 317 |
"bos_token": "<|endoftext|>",
|
| 318 |
+
"chat_template": "{% for message in messages %}{% if message['role'] == 'system' %}{{'<|system|>\n' + message['content'] + '<|end|>\n'}}{% elif message['role'] == 'user' %}{{'<|user|>\n' + message['content'] + '<|end|>\n'}}{% elif message['role'] == 'assistant' %}{{'<|assistant|>\n' + message['content'] + '<|end|>\n'}}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|assistant|>\n' }}{% else %}{{ eos_token }}{% endif %}",
|
| 319 |
"clean_up_tokenization_spaces": true,
|
| 320 |
"eos_token": "<|endoftext|>",
|
| 321 |
"model_max_length": 2048,
|