tomer-deci
commited on
Commit
•
fb62397
1
Parent(s):
49422ae
added chat_template
Browse files- tokenizer_config.json +3 -2
tokenizer_config.json
CHANGED
@@ -36,5 +36,6 @@
|
|
36 |
"spaces_between_special_tokens": false,
|
37 |
"tokenizer_class": "LlamaTokenizer",
|
38 |
"unk_token": "<unk>",
|
39 |
-
"use_default_system_prompt": true
|
40 |
-
}
|
|
|
|
36 |
"spaces_between_special_tokens": false,
|
37 |
"tokenizer_class": "LlamaTokenizer",
|
38 |
"unk_token": "<unk>",
|
39 |
+
"use_default_system_prompt": true,
|
40 |
+
"chat_template": "{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '### User:\n' + message['content'] }}\n{% elif message['role'] == 'system' %}\n{{ '### System:\n' + message['content'] }}\n{% elif message['role'] == 'assistant' %}\n{{ '### Assistant:\n' + message['content'] }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '### Assistant:' }}\n{% endif %}\n{% endfor %}"
|
41 |
+
}
|