groupuser/stablelm-2-1_6b
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: add chat template to tokenizer config

Browse Source
This commit is contained in:
Max 2024-01-19 22:01:18 +00:00 committed by system
parent e6185e1580
commit 507254d7bd
No known key found for this signature in database
GPG Key ID: 6A528E38E0733467
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tokenizer_config.json
View File

@ -5,5 +5,13 @@
"tokenization_arcade100k.Arcade100kTokenizer",
null
]
}
},
"added_tokens_decoder": {},
"bos_token": "<|endoftext|>",
"chat_template": "{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '<|user|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'system' %}\n{{ '<|system|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'assistant' %}\n{{ '<|assistant|>\n' + message['content'] + eos_token }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '<|assistant|>' }}\n{% endif %}\n{% endfor %}",
"clean_up_tokenization_spaces": true,
"eos_token": "<|endoftext|>",
"errors": "replace",
"model_max_length": 4096,
"pad_token": "<|endoftext|>"
}