simplify
Browse files- chat_template.jinja +4 -13
chat_template.jinja
CHANGED
|
@@ -1,22 +1,13 @@
|
|
|
|
|
| 1 |
{%- if messages[0]["role"] == "system" -%}
|
| 2 |
-
{{- "<|
|
| 3 |
-
{{- messages[0]["content"] -}}
|
| 4 |
-
{{- "\n\n" -}}
|
| 5 |
-
{{- messages[1]["content"] -}}
|
| 6 |
-
{{- "<|user_end|>" -}}
|
| 7 |
{%- set messages = messages[2:] -%}
|
| 8 |
-
{%- else -%}
|
| 9 |
-
{{- "<|bos|>" -}}
|
| 10 |
{%- endif -%}
|
| 11 |
{%- for message in messages -%}
|
| 12 |
{%- if loop.index0 % 2 == 0 -%}
|
| 13 |
-
{{- "<|user_start|>" -}}
|
| 14 |
-
{{- message["content"] -}}
|
| 15 |
-
{{- "<|user_end|>" -}}
|
| 16 |
{%- else -%}
|
| 17 |
-
{{- "<|assistant_start|>" -}}
|
| 18 |
-
{{- message["content"] -}}
|
| 19 |
-
{{- "<|assistant_end|>" -}}
|
| 20 |
{%- endif -%}
|
| 21 |
{%- endfor -%}
|
| 22 |
{%- if add_generation_prompt -%}
|
|
|
|
| 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 -%}
|