Create chat template file (#5)
Browse files- Create chat template file (8998cdf6d5fe074d2886f747f1d69fac9e198d3b)
- simplify (5cb28c9b8367f8985b2f2d7deaef5ebbf0579109)
Co-authored-by: Joshua <[email protected]>
- chat_template.jinja +15 -0
chat_template.jinja
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{{- "<|bos|>" -}}
|
| 2 |
+
{%- if messages[0]["role"] == "system" -%}
|
| 3 |
+
{{- "<|user_start|>" + messages[0]["content"] + "\n\n" + messages[1]["content"] + "<|user_end|>" -}}
|
| 4 |
+
{%- set messages = messages[2:] -%}
|
| 5 |
+
{%- endif -%}
|
| 6 |
+
{%- for message in messages -%}
|
| 7 |
+
{%- if loop.index0 % 2 == 0 -%}
|
| 8 |
+
{{- "<|user_start|>" + message["content"] + "<|user_end|>" -}}
|
| 9 |
+
{%- else -%}
|
| 10 |
+
{{- "<|assistant_start|>" + message["content"] + "<|assistant_end|>" -}}
|
| 11 |
+
{%- endif -%}
|
| 12 |
+
{%- endfor -%}
|
| 13 |
+
{%- if add_generation_prompt -%}
|
| 14 |
+
{{- "<|assistant_start|>" -}}
|
| 15 |
+
{%- endif -%}
|