awni commited on
Commit
3d236a6
·
verified ·
1 Parent(s): 5a55049

Update chat_template.jinja

Browse files
Files changed (1) hide show
  1. 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
- {{ "<|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 -%}
 
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 -%}