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,
|