pandora-s commited on
Commit
8d87519
·
verified ·
1 Parent(s): 2725a4f

add template

Browse files
Files changed (1) hide show
  1. tokenizer_config.json +1 -0
tokenizer_config.json CHANGED
@@ -6174,6 +6174,7 @@
6174
  },
6175
  "bos_token": "<s>",
6176
  "clean_up_tokenization_spaces": false,
 
6177
  "eos_token": "</s>",
6178
  "legacy": true,
6179
  "model_max_length": 1000000000000000019884624838656,
 
6174
  },
6175
  "bos_token": "<s>",
6176
  "clean_up_tokenization_spaces": false,
6177
+ "chat_template": "{{ bos_token }}{% for message in messages %}{% if message['role'] == 'user' %}{{ '[INST] ' + message['content'] + '[/INST]' }}{% elif message['role'] == 'system' %}{{ '[SYSTEM_PROMPT] ' + message['content'] + '[/SYSTEM_PROMPT]' }}{% elif message['role'] == 'assistant' %}{{ ' ' + message['content'] + eos_token }}{% else %}{{ raise_exception('Only user, system and assistant roles are supported!') }}{% endif %}{% endfor %}"
6178
  "eos_token": "</s>",
6179
  "legacy": true,
6180
  "model_max_length": 1000000000000000019884624838656,