JorgeDeC commited on
Commit
ac92904
·
verified ·
1 Parent(s): fa4d4e3

Update chat_template

Browse files
Files changed (1) hide show
  1. tokenizer_config.json +1 -1
tokenizer_config.json CHANGED
@@ -2071,7 +2071,7 @@
2071
  ],
2072
  "bos_token": "<|im_start|>",
2073
  "clean_up_tokenization_spaces": true,
2074
- "chat_template": "{% if not add_generation_prompt is defined %}{% set add_generation_prompt = false %}{% endif %}{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
2075
  "eos_token": "<|im_end|>",
2076
  "model_input_names": [
2077
  "input_ids",
 
2071
  ],
2072
  "bos_token": "<|im_start|>",
2073
  "clean_up_tokenization_spaces": true,
2074
+ "chat_template": "{% if not add_generation_prompt is defined %}{% set add_generation_prompt = false %}{% endif %}{{ bos_token }}{% for message in messages %}{% if message['role'] == 'user' %}{{ 'User: ' + message['content'] + '\n\n' }}{% elif message['role'] == 'assistant' %}{{ 'Assistant: ' + message['content'] + eos_token }}{% elif message['role'] == 'system' %}{{ message['content'] + '\n\n' }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ 'Assistant:' }}{% endif %}",
2075
  "eos_token": "<|im_end|>",
2076
  "model_input_names": [
2077
  "input_ids",