Daehyun Ahn commited on
Commit
7758bc9
·
verified ·
1 Parent(s): 1830c1b

add chat_template to tokenizer

Browse files
Files changed (1) hide show
  1. tokenizer_config.json +1 -0
tokenizer_config.json CHANGED
@@ -315,6 +315,7 @@
315
  }
316
  },
317
  "bos_token": "<|endoftext|>",
 
318
  "clean_up_tokenization_spaces": true,
319
  "eos_token": "<|endoftext|>",
320
  "model_max_length": 2048,
 
315
  }
316
  },
317
  "bos_token": "<|endoftext|>",
318
+ "chat_template": "{% for message in messages %}{% if message['role'] == 'system' %}{{'<|system|>\n' + message['content'] + '<|end|>\n'}}{% elif message['role'] == 'user' %}{{'<|user|>\n' + message['content'] + '<|end|>\n'}}{% elif message['role'] == 'assistant' %}{{'<|assistant|>\n' + message['content'] + '<|end|>\n'}}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|assistant|>\n' }}{% else %}{{ eos_token }}{% endif %}",
319
  "clean_up_tokenization_spaces": true,
320
  "eos_token": "<|endoftext|>",
321
  "model_max_length": 2048,