maywell commited on
Commit
2a8d722
1 Parent(s): aa69c3e

fix_template

Browse files
Files changed (1) hide show
  1. tokenizer_config.json +1 -1
tokenizer_config.json CHANGED
@@ -2061,7 +2061,7 @@
2061
  "<|none1|>"
2062
  ],
2063
  "bos_token": "<|begin_of_text|>",
2064
- "chat_template": "{% if messages[0]['role'] == 'system' %}{% set system_message = messages[0]['content'] %}{% endif %}{% if system_message is defined %}{{ system_message }}{% endif %}{% for message in messages %}{% set content = message['content'] %}{% if message['role'] == 'user' %}{{ content }}{% elif message['role'] == 'assistant' %}{{ content + '\\n' }}{% endif %}{% endfor %}",
2065
  "clean_up_tokenization_spaces": true,
2066
  "eos_token": "<|end_of_text|>",
2067
  "model_input_names": [
 
2061
  "<|none1|>"
2062
  ],
2063
  "bos_token": "<|begin_of_text|>",
2064
+ "chat_template": "{% if messages[0]['role'] == 'system' %}{% set loop_messages = messages[1:] %}{% set system_message = messages[0]['content'] %}{% else %}{% set loop_messages = messages %}{% set system_message = 'You are a helpful assistant.' %}{% endif %}{% if not add_generation_prompt is defined %}{% set add_generation_prompt = false %}{% endif %}{% for message in loop_messages %}{% if loop.index0 == 0 %}{{'<|im_start|>system\n' + system_message + '<|im_end|>\n'}}{% endif %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
2065
  "clean_up_tokenization_spaces": true,
2066
  "eos_token": "<|end_of_text|>",
2067
  "model_input_names": [