Add baichuan chat template jinjia file (#2390)

74cd5abd · KKY · GitHub · 28c3f121 · 74cd5abd
Unverified Commit 74cd5abd authored Jan 09, 2024 by KKY Committed by GitHub Jan 09, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 22 additions and 0 deletions

examples/template_baichuan.jinja examples/template_baichuan.jinja +22 -0

No files found.
--- a/examples/template_baichuan.jinja
+++ b/examples/template_baichuan.jinja
+{{ (messages|selectattr('role', 'equalto', 'system')|list|last).content|trim if (messages|selectattr('role', 'equalto', 'system')|list) else '' }}
+
+{% for message in messages %}
+{% if message['role'] == 'user' %}
+<reserved_106>
+{{ message['content']|trim -}}
+{% if not loop.last %}
+
+
+{% endif %}
+{% elif message['role'] == 'assistant' %}
+<reserved_107>
+{{ message['content']|trim -}}
+{% if not loop.last %}
+
+
+{% endif %}
+{% endif %}
+{% endfor %}
+{% if add_generation_prompt and messages[-1]['role'] != 'assistant' %}
+<reserved_107>
+{% endif %}
\ No newline at end of file