Skip to content

GitLab

Unverified Commit cf6ff182 authored by Dinghow Yang's avatar Dinghow Yang Committed by GitHub
Browse files

Fix Baichuan chat template (#3340)

parent 14e3f9a1
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 19 deletions
examples/template_baichuan.jinja
View file @ cf6ff182
{{ (messages|selectattr('role', 'equalto', 'system')|list|last).content|trim if (messages|selectattr('role', 'equalto', 'system')|list) else '' }} {{ (messages|selectattr('role', 'equalto', 'system')|list|last).content|trim if (messages|selectattr('role', 'equalto', 'system')|list) else '' }}
{% for message in messages %} {%- for message in messages -%}
{% if message['role'] == 'user' %} {%- if message['role'] == 'user' -%}
<reserved_106> {{- '<reserved_106>' + message['content'] -}}
{{ message['content']|trim -}} {%- elif message['role'] == 'assistant' -%}
{% if not loop.last %} {{- '<reserved_107>' + message['content'] -}}
{%- endif -%}
{%- endfor -%}
{% endif %}
{% elif message['role'] == 'assistant' %} {%- if add_generation_prompt and messages[-1]['role'] != 'assistant' -%}
<reserved_107> {{- '<reserved_107>' -}}
{{ message['content']|trim -}}
{% if not loop.last %}
{% endif %}
{% endif %}
{% endfor %}
{% if add_generation_prompt and messages[-1]['role'] != 'assistant' %}
<reserved_107>
{% endif %} {% endif %}
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment