test commited on
Commit
8189385
1 Parent(s): d02e4aa

update chat template

Browse files
Files changed (1) hide show
  1. tokenizer_config.json +1 -1
tokenizer_config.json CHANGED
@@ -41,5 +41,5 @@
41
  "tokenizer_class": "LlamaTokenizer",
42
  "unk_token": "<unk>",
43
  "use_default_system_prompt": true,
44
- "chat_template": "{%- if messages[0]['role'] == 'system' %}\n {%- set system_message = messages[0]['content'] %}\n {%- set loop_messages = messages[1:] %}\n{%- else %}\n {%- set loop_messages = messages %}\n{%- endif %}\n\n{{- bos_token }}\n{%- for message in loop_messages %}\n {%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}\n {{- raise_exception('After the optional system message, conversation roles must alternate user/assistant/user/assistant/...') }}\n {%- endif %}\n {%- if loop.first and system_message is defined %}\n {{- ' ### System:\n ' + system_message + '\n' }}\n {%- endif %}\n {%- if message['role'] == 'user' %}\n {{- ' ### User:\n ' + message['content'] + ' \n' }}\n {%- elif message['role'] == 'assistant' %}\n {{- ' ### Assistant:\n ' + message['content'] + eos_token + '\n'}}\n {%- else %}\n {{- raise_exception('Only user and assistant roles are supported, with the exception of an initial optional system message!') }}\n {%- endif %}\n{%- endfor %}{% if add_generation_prompt %}{{ ' ### Assistant:\n' }}{% endif %}\n"
45
  }
 
41
  "tokenizer_class": "LlamaTokenizer",
42
  "unk_token": "<unk>",
43
  "use_default_system_prompt": true,
44
+ "chat_template": "{%- if messages[0]['role'] == 'system' %}\n {%- set system_message = messages[0]['content'] %}\n {%- set loop_messages = messages[1:] %}\n{%- else %}\n {%- set loop_messages = messages %}\n{%- endif %}\n\n{%- for message in loop_messages %}\n {%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}\n {{- raise_exception('After the optional system message, conversation roles must alternate user/assistant/user/assistant/...') }}\n {%- endif %}\n {%- if loop.first and system_message is defined %}\n {{- '### System:\n' + system_message + '\n' }}\n {%- endif %}\n {%- if message['role'] == 'user' %}\n {{- '### User:\n' + message['content'] + ' \n' }}\n {%- elif message['role'] == 'assistant' %}\n {{- '### Assistant:\n' + message['content'] + eos_token + '\n'}}\n {%- else %}\n {{- raise_exception('Only user and assistant roles are supported, with the exception of an initial optional system message!') }}\n {%- endif %}\n{%- endfor %}{% if add_generation_prompt %}{{ '### Assistant:\n' }}{% endif %}\n"
45
  }