ivoschaper commited on
Commit
e9d7558
1 Parent(s): bea54ae

add support for `add_generation_prompt` parameter

Browse files
Files changed (1) hide show
  1. tokenizer_config.json +1 -1
tokenizer_config.json CHANGED
@@ -6594,7 +6594,7 @@
6594
  }
6595
  },
6596
  "bos_token": "<|begin_of_text|>",
6597
- "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' }}",
6598
  "eos_token": "<|endoftext|>",
6599
  "pad_token": "<|padding|>",
6600
  "unk_token": "<|unknown|>",
 
6594
  }
6595
  },
6596
  "bos_token": "<|begin_of_text|>",
6597
+ "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 %}",
6598
  "eos_token": "<|endoftext|>",
6599
  "pad_token": "<|padding|>",
6600
  "unk_token": "<|unknown|>",