Update chat_template.jinja

chat_template.jinja

@@ -7,40 +7,16 @@
 {%- if not sep_token is defined -%}
     {%- set sep_token = "<|end_of_sentence|>" -%}
 {%- endif -%}
-{%- if not image_token is defined -%}
-    {%- set image_token = "<|IMAGE_START|><|IMAGE_PLACEHOLDER|><|IMAGE_END|>" -%}
-{%- endif -%}
 {{- cls_token -}}
 {%- for message in messages -%}
     {%- if message["role"] == "user" -%}
-        {{- "User: " -}}
-        {%- for content in message["content"] -%}
-            {%- if content["type"] == "image" -%}
-                {{ image_token }}
-            {%- endif -%}
-        {%- endfor -%}
-        {%- for content in message["content"] -%}
-            {%- if content["type"] == "text" -%}
-                {{ content["text"] }}
-            {%- endif -%}
-        {%- endfor -%}
-        {{ "\n" -}}
+        {{- "User: <|IMAGE_START|><|IMAGE_PLACEHOLDER|><|IMAGE_END|>" + message["content"] + "\n" -}}
     {%- elif message["role"] == "assistant" -%}
-        {{- "Assistant: " -}}
-        {%- for content in message["content"] -%}
-            {%- if content["type"] == "text" -%}
-                {{ content["text"] + "\n" }}
-            {%- endif -%}
-        {%- endfor -%}
-        {{ sep_token -}}
+        {{- "Assistant: " + message["content"] + sep_token -}}
     {%- elif message["role"] == "system" -%}
-        {%- for content in message["content"] -%}
-            {%- if content["type"] == "text" -%}
-                {{ content["text"] + "\n" }}
-            {%- endif -%}
-        {%- endfor -%}
+        {{- message["content"] -}}
     {%- endif -%}
 {%- endfor -%}
 {%- if add_generation_prompt -%}
     {{- "Assistant: " -}}
-{%- endif -%}
+{%- endif -%}