File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed
Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change 44
55namespace Codewithkyrian \Transformers \PreTrainedTokenizers ;
66
7- class GemmaTokenizer extends PretrainedTokenizer
7+ class GemmaTokenizer extends PreTrainedTokenizer
88{
99 protected string $ defaultChatTemplate = "{% if messages[0]['role'] == 'system' %}{{ raise_exception('System role not supported') }}{% endif %}{% for message in messages %}{% if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}{{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }}{% endif %}{% if (message['role'] == 'assistant') %}{% set role = 'model' %}{% else %}{% set role = message['role'] %}{% endif %}{{ '<start_of_turn>' + role + ' \n' + message['content'] | trim + '<end_of_turn> \n' }}{% endfor %}{% if add_generation_prompt %}{{'<start_of_turn>model \n'}}{% endif %} " ;
1010}
You can’t perform that action at this time.
0 commit comments