emilss commited on
Commit
7c4e38c
·
1 Parent(s): 63aba4a

Add chat template

Browse files
Files changed (1) hide show
  1. tokenizer_config.json +1 -0
tokenizer_config.json CHANGED
@@ -38,6 +38,7 @@
38
  },
39
  "bos_token": "<s>",
40
  "clean_up_tokenization_spaces": false,
 
41
  "eos_token": "</s>",
42
  "extra_special_tokens": {},
43
  "legacy": false,
 
38
  },
39
  "bos_token": "<s>",
40
  "clean_up_tokenization_spaces": false,
41
+ "chat_template": "{% if messages[0]['role'] == 'system' %}\n {% set system_message = messages[0]['content'] | trim + '\n\n' %}\n {% set messages = messages[1:] %}\n{% else %}\n {% set system_message = '' %}\n{% endif %}\n\n{{ bos_token + system_message }}\n{% for message in messages %}\n {% if message['role'] == 'system' %}\n {{ '### System:\n' + message['content'] | trim + '\n\n' }}\n {% elif message['role'] == 'user' %}\n {{ '### Instruction:\n' + message['content'] | trim + '\n\n' }}\n {% elif message['role'] == 'assistant' %}\n {{ '### Response:\n' + message['content'] | trim + eos_token + '\n\n' }}\n {% endif %}\n{% endfor %}\n{% if add_generation_prompt %}\n {{ '### Response:\n' }}\n{% endif %}",
42
  "eos_token": "</s>",
43
  "extra_special_tokens": {},
44
  "legacy": false,