Commit
•
1cfbbfd
1
Parent(s):
6d10497
增加對tokenizer.chat_template的支援 (#22)
Browse files- Update tokenizer_config.json (8c4bd56f3f760417ee95b6d20b9f78302f80e78d)
Co-authored-by: Philip Huang <[email protected]>
- tokenizer_config.json +1 -0
tokenizer_config.json
CHANGED
@@ -3,6 +3,7 @@
|
|
3 |
"remove_space": false,
|
4 |
"do_lower_case": false,
|
5 |
"tokenizer_class": "ChatGLMTokenizer",
|
|
|
6 |
"auto_map": {
|
7 |
"AutoTokenizer": [
|
8 |
"tokenization_chatglm.ChatGLMTokenizer",
|
|
|
3 |
"remove_space": false,
|
4 |
"do_lower_case": false,
|
5 |
"tokenizer_class": "ChatGLMTokenizer",
|
6 |
+
"chat_template": "{% for message in messages %}{% if loop.first %}[gMASK]sop<|{{ message['role'] }}|> \n {{ message['content'] }}{% else %}<|{{ message['role'] }}|> \n {{ message['content'] }}{% endif %}{% endfor %}{% if add_generation_prompt %}<|assistant|>{% endif %}",
|
7 |
"auto_map": {
|
8 |
"AutoTokenizer": [
|
9 |
"tokenization_chatglm.ChatGLMTokenizer",
|