rajammanabrolu commited on
Commit
e30d950
·
1 Parent(s): a755aa7

Update tiktoken.py

Browse files
Files changed (1) hide show
  1. tiktoken.py +1 -1
tiktoken.py CHANGED
@@ -215,7 +215,7 @@ class TiktokenTokenizerWrapper(PreTrainedTokenizer):
215
  '{% if (add_generation_prompt == true) %}'
216
  "{{'<|im_start|>' + 'assistant' + '\n'}}"
217
  "{% else %}"
218
- "{{ eos_token }}"
219
  template = template.replace(
220
  'USE_DEFAULT_PROMPT',
221
  'true' if self.use_default_system_prompt else 'false')
 
215
  '{% if (add_generation_prompt == true) %}'
216
  "{{'<|im_start|>' + 'assistant' + '\n'}}"
217
  "{% else %}"
218
+ "{{ eos_token }}")
219
  template = template.replace(
220
  'USE_DEFAULT_PROMPT',
221
  'true' if self.use_default_system_prompt else 'false')