Add chat_template into the tokenizer
Browse files- tokenizer_config.json +2 -1
tokenizer_config.json
CHANGED
@@ -7,5 +7,6 @@
|
|
7 |
"pad_token": "<pad>",
|
8 |
"padding_side": "right",
|
9 |
"tokenizer_class": "BloomTokenizer",
|
10 |
-
"unk_token": "<unk>"
|
|
|
11 |
}
|
|
|
7 |
"pad_token": "<pad>",
|
8 |
"padding_side": "right",
|
9 |
"tokenizer_class": "BloomTokenizer",
|
10 |
+
"unk_token": "<unk>",
|
11 |
+
"chat_template": "{% if not add_generation_prompt is defined %}{% set add_generation_prompt = false %}{% endif %}{% for message in messages %}{% if message['role'] == 'user' %}{{ '### Câu hỏi: ' + message['content'].strip() }}{% elif message['role'] == 'assistant' %}{{ '\n### Trả lời: ' + message['content'] + eos_token }}{% endif %}{% if loop.last %}{% if message['role'] == 'user' and add_generation_prompt %}{{ '\n### Trả lời:' }}{% endif %}{% endif %}{% endfor %}"
|
12 |
}
|