Add chat template
Browse files- tokenizer_config.json +1 -0
tokenizer_config.json
CHANGED
|
@@ -38,6 +38,7 @@
|
|
| 38 |
},
|
| 39 |
"bos_token": "<s>",
|
| 40 |
"clean_up_tokenization_spaces": false,
|
|
|
|
| 41 |
"eos_token": "</s>",
|
| 42 |
"extra_special_tokens": {},
|
| 43 |
"legacy": false,
|
|
|
|
| 38 |
},
|
| 39 |
"bos_token": "<s>",
|
| 40 |
"clean_up_tokenization_spaces": false,
|
| 41 |
+
"chat_template": "{% if messages[0]['role'] == 'system' %}\n {% set system_message = messages[0]['content'] | trim + '\n\n' %}\n {% set messages = messages[1:] %}\n{% else %}\n {% set system_message = '' %}\n{% endif %}\n\n{{ bos_token + system_message }}\n{% for message in messages %}\n {% if message['role'] == 'system' %}\n {{ '### System:\n' + message['content'] | trim + '\n\n' }}\n {% elif message['role'] == 'user' %}\n {{ '### Instruction:\n' + message['content'] | trim + '\n\n' }}\n {% elif message['role'] == 'assistant' %}\n {{ '### Response:\n' + message['content'] | trim + eos_token + '\n\n' }}\n {% endif %}\n{% endfor %}\n{% if add_generation_prompt %}\n {{ '### Response:\n' }}\n{% endif %}",
|
| 42 |
"eos_token": "</s>",
|
| 43 |
"extra_special_tokens": {},
|
| 44 |
"legacy": false,
|