Update chat_template.jinja
Browse files- chat_template.jinja +4 -4
chat_template.jinja
CHANGED
|
@@ -50,16 +50,16 @@
|
|
| 50 |
{%- for message in messages -%}
|
| 51 |
{%- set content = message.content if message.content is string else "" -%}
|
| 52 |
{%- if message.role == "user" -%}
|
| 53 |
-
{{ "<|im_start|>" + message.role + "\n" + content + "<|im_end|>\n" }}
|
| 54 |
{%- elif message.role == "assistant" -%}
|
| 55 |
{% generation %}
|
| 56 |
-
{{ "<|im_start|>assistant\n" + content.lstrip("\n") + "<|im_end|>\n" }}
|
| 57 |
{% endgeneration %}
|
| 58 |
{%- elif message.role == "tool" -%}
|
| 59 |
-
|
| 60 |
{%- endif -%}
|
| 61 |
{%- endfor -%}
|
| 62 |
{# ───── generation prompt ───── #}
|
| 63 |
{%- if add_generation_prompt -%}
|
| 64 |
-
{{ "<|im_start|>assistant\n" }}
|
| 65 |
{%- endif -%}
|
|
|
|
| 50 |
{%- for message in messages -%}
|
| 51 |
{%- set content = message.content if message.content is string else "" -%}
|
| 52 |
{%- if message.role == "user" -%}
|
| 53 |
+
{{- "<|im_start|>" + message.role + "\n" + content + "<|im_end|>\n" }}
|
| 54 |
{%- elif message.role == "assistant" -%}
|
| 55 |
{% generation %}
|
| 56 |
+
{{- "<|im_start|>assistant\n" + content.lstrip("\n") + "<|im_end|>\n" }}
|
| 57 |
{% endgeneration %}
|
| 58 |
{%- elif message.role == "tool" -%}
|
| 59 |
+
{{- "<|im_start|>" + "user\n" + content + "<|im_end|>\n" }}
|
| 60 |
{%- endif -%}
|
| 61 |
{%- endfor -%}
|
| 62 |
{# ───── generation prompt ───── #}
|
| 63 |
{%- if add_generation_prompt -%}
|
| 64 |
+
{{- "<|im_start|>assistant\n" }}
|
| 65 |
{%- endif -%}
|