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

Update chat_template.jinja

Browse files
Files changed (1) hide show
  1. chat_template.jinja +1 -3
chat_template.jinja CHANGED
@@ -50,11 +50,9 @@
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 -%}
 
50
  {%- for message in messages -%}
51
  {%- set content = message.content if message.content is string else "" -%}
52
  {%- if message.role == "user" -%}
53
+ {{- "<|im_start|>" + "user\n" + content + "<|im_end|>\n" }}
54
  {%- elif message.role == "assistant" -%}
 
55
  {{- "<|im_start|>assistant\n" + content.lstrip("\n") + "<|im_end|>\n" }}
 
56
  {%- elif message.role == "tool" -%}
57
  {{- "<|im_start|>" + "user\n" + content + "<|im_end|>\n" }}
58
  {%- endif -%}