BusRune commited on
Commit
c515233
·
verified ·
1 Parent(s): 1d2a70a

Corrected the template.

Browse files
Files changed (1) hide show
  1. tokenizer_config.json +1 -1
tokenizer_config.json CHANGED
@@ -66,7 +66,7 @@
66
  "▁<EOT>"
67
  ],
68
  "bos_token": "<s>",
69
- "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 = false %}{% endif %}{% for message in loop_messages %}{% if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}{{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }}{% endif %}{% if loop.index0 == 0 and system_message != false %}{% set content = '<<SYS>>\\n' + system_message + '\\n<</SYS>>\\n\\n' + message['content'] %}{% else %}{% set content = message['content'] %}{% endif %}{% if message['role'] == 'user' %}{{ bos_token + '[INST] ' + content | trim + ' [/INST]' }}{% elif message['role'] == 'assistant' %}{{ ' ' + content | trim + ' ' + eos_token }}{% endif %}{% endfor %}",
70
  "clean_up_tokenization_spaces": false,
71
  "eos_token": "</s>",
72
  "eot_token": "▁<EOT>",
 
66
  "▁<EOT>"
67
  ],
68
  "bos_token": "<s>",
69
+ "chat_template": "{% for message in messages %}{% if loop.first and messages[0]['role'] != 'system' %}{{ '<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n' }}{% endif %}{{'<|im_start|>' + message['role'] + (' [' + message['name'] + ']' if message['name'] else '') + '\n' + message['content'] + ('<|im_end|>' if message['role'] != 'assistant' or not loop.last else '') + '\n'}}{% endfor %}{% if messages[-1]['role'] != 'assistant' %}{{ '<|im_start|>assistant\n' }}{% endif %}",
70
  "clean_up_tokenization_spaces": false,
71
  "eos_token": "</s>",
72
  "eot_token": "▁<EOT>",