tomer-deci
commited on
Commit
•
8d9cdb8
1
Parent(s):
83e1abe
added chat_template
Browse files- config.json +1 -1
- 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 |
+
}
|