tomer-deci commited on
Commit
8d9cdb8
1 Parent(s): 83e1abe

added chat_template

Browse files
Files changed (2) hide show
  1. config.json +1 -1
  2. tokenizer_config.json +3 -2
config.json CHANGED
@@ -27,4 +27,4 @@
27
  "use_cache": true,
28
  "vocab_size": 32000,
29
  "tokenizer_class": "LlamaTokenizer"
30
- }
 
27
  "use_cache": true,
28
  "vocab_size": 32000,
29
  "tokenizer_class": "LlamaTokenizer"
30
+ }
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
+ }