Update chat_template.jinja
Browse files- chat_template.jinja +8 -2
chat_template.jinja
CHANGED
|
@@ -1,4 +1,10 @@
|
|
| 1 |
{%- for message in messages -%}
|
| 2 |
-
{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
{%- endfor -%}
|
| 4 |
-
{{ '
|
|
|
|
| 1 |
{%- for message in messages -%}
|
| 2 |
+
{%- if loop.first and message.role == 'user' -%}
|
| 3 |
+
{{ '<user> ' + message.content + ' ' }}
|
| 4 |
+
{%- elif message.role == 'user' -%}
|
| 5 |
+
{{ '<user> ' + message.content + ' ' }}
|
| 6 |
+
{%- elif message.role == 'assistant' -%}
|
| 7 |
+
{{ '<bot> ' + message.content + ' ' }}
|
| 8 |
+
{%- endif -%}
|
| 9 |
{%- endfor -%}
|
| 10 |
+
{{ '<bot>' }}
|