remove extra whitespace from replacements

Signed-off-by: Jared Van Bortel <jared@nomic.ai>
This commit is contained in:
Jared Van Bortel 2025-02-12 14:21:12 -05:00
parent 0ce0189b97
commit e9f5d15c09

View File

@ -113,7 +113,7 @@ const std::unordered_map<std::string_view, std::string_view> CHAT_TEMPLATE_SUBST
{%- elif message['role'] == 'system' %} {%- elif message['role'] == 'system' %}
{{- '<|system|>\n' + message['content'] + eos_token }} {{- '<|system|>\n' + message['content'] + eos_token }}
{%- elif message['role'] == 'assistant' %} {%- elif message['role'] == 'assistant' %}
{{- '<|assistant|>\n' + message['content'] + eos_token }} {{- '<|assistant|>\n' + message['content'] + eos_token }}
{%- endif %} {%- endif %}
{%- if loop.last and add_generation_prompt %} {%- if loop.last and add_generation_prompt %}
{{- '<|assistant|>' }} {{- '<|assistant|>' }}
@ -660,9 +660,9 @@ const std::unordered_map<std::string_view, std::string_view> CHAT_TEMPLATE_SUBST
{{- '<|user|>\n' + message['content'] + '\n' }} {{- '<|user|>\n' + message['content'] + '\n' }}
{%- elif message['role'] == 'assistant' %} {%- elif message['role'] == 'assistant' %}
{%- if not loop.last %} {%- if not loop.last %}
{{- '<|assistant|>\n' + message['content'] + eos_token + '\n' }} {{- '<|assistant|>\n' + message['content'] + eos_token + '\n' }}
{%- else %} {%- else %}
{{- '<|assistant|>\n' + message['content'] + eos_token }} {{- '<|assistant|>\n' + message['content'] + eos_token }}
{%- endif %} {%- endif %}
{%- endif %} {%- endif %}
{%- if loop.last and add_generation_prompt %} {%- if loop.last and add_generation_prompt %}