tomer-deci commited on
Commit
fb62397
1 Parent(s): 49422ae

added chat_template

Browse files
Files changed (1) hide show
  1. tokenizer_config.json +3 -2
tokenizer_config.json CHANGED
@@ -36,5 +36,6 @@
36
  "spaces_between_special_tokens": false,
37
  "tokenizer_class": "LlamaTokenizer",
38
  "unk_token": "<unk>",
39
- "use_default_system_prompt": true
40
- }
 
 
36
  "spaces_between_special_tokens": false,
37
  "tokenizer_class": "LlamaTokenizer",
38
  "unk_token": "<unk>",
39
+ "use_default_system_prompt": true,
40
+ "chat_template": "{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '### User:\n' + message['content'] }}\n{% elif message['role'] == 'system' %}\n{{ '### System:\n' + message['content'] }}\n{% elif message['role'] == 'assistant' %}\n{{ '### Assistant:\n' + message['content'] }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '### Assistant:' }}\n{% endif %}\n{% endfor %}"
41
+ }