dqnguyen commited on
Commit
191b9a4
1 Parent(s): d839883

Add chat_template into the tokenizer

Browse files
Files changed (1) hide show
  1. 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
  }