Add chat_template to tokenizer
#5
by
sean-public
- opened
- tokenizer_config.json +1 -0
tokenizer_config.json
CHANGED
@@ -9,6 +9,7 @@
|
|
9 |
"rstrip": false,
|
10 |
"single_word": false
|
11 |
},
|
|
|
12 |
"clean_up_tokenization_spaces": false,
|
13 |
"eos_token": {
|
14 |
"__type": "AddedToken",
|
|
|
9 |
"rstrip": false,
|
10 |
"single_word": false
|
11 |
},
|
12 |
+
"chat_template": "{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '<|user|>\n' + message['content'] }}\n{% elif message['role'] == 'assistant' %}\n{{ '<|assistant|>\n' + message['content'] + eos_token }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '<|assistant|>' }}\n{% endif %}\n{% endfor %}",
|
13 |
"clean_up_tokenization_spaces": false,
|
14 |
"eos_token": {
|
15 |
"__type": "AddedToken",
|