AnatoliiPotapov amnn314 commited on
Commit
d346cb6
1 Parent(s): a7f8ed0

Fix ignored 'add_generation_prompt' in the chat template (#4)

Browse files

- Fix ignored 'add_generation_prompt' in the chat template (813d98c020948768b6c1ba806cf6c387cb57a999)


Co-authored-by: amnn314 <amnn314@users.noreply.huggingface.co>

Files changed (1) hide show
  1. tokenizer_config.json +1 -1
tokenizer_config.json CHANGED
@@ -2074,7 +2074,7 @@
2074
  }
2075
  },
2076
  "bos_token": "<|begin_of_text|>",
2077
- "chat_template": "{% set loop_messages = messages %}{% for message in loop_messages %}{% set content = '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'+ message['content'] | trim + '<|eot_id|>' %}{% if loop.index0 == 0 %}{% set content = bos_token + content %}{% endif %}{{ content }}{% endfor %}{{ '<|start_header_id|>assistant<|end_header_id|>\n\n' }}",
2078
  "clean_up_tokenization_spaces": true,
2079
  "eos_token": "<|end_of_text|>",
2080
  "model_input_names": [
 
2074
  }
2075
  },
2076
  "bos_token": "<|begin_of_text|>",
2077
+ "chat_template": "{% set loop_messages = messages %}{% for message in loop_messages %}{% set content = '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'+ message['content'] | trim + '<|eot_id|>' %}{% if loop.index0 == 0 %}{% set content = bos_token + content %}{% endif %}{{ content }}{% endfor %}{% if add_generation_prompt %}{{ '<|start_header_id|>assistant<|end_header_id|>\n\n' }}{% endif %}",
2078
  "clean_up_tokenization_spaces": true,
2079
  "eos_token": "<|end_of_text|>",
2080
  "model_input_names": [