Update magistral_chat_template.jinja

magistral_chat_template.jinja

@@ -54,7 +54,7 @@
 {#- Handle conversation messages. #}
 {%- for message in loop_messages %}
 
-    {#- User messages supports text content. #}
+    {#- User messages support text and image content. #}
     {%- if message['role'] == 'user' %}
         {%- if message['content'] is string %}
             {{- '[INST]' + message['content'] + '[/INST]' }}
@@ -63,8 +63,10 @@
             {%- for block in message['content'] %}
                 {%- if block['type'] == 'text' %}
                     {{- block['text'] }}
+                {%- elif block['type'] == 'image' or block['type'] == 'image_url' %}
+                    {{- '[IMG]' }}
                 {%- else %}
-                    {{- raise_exception('Only text chunks are supported in user message content.') }}
+                    {{- raise_exception('Only text and image chunks are supported in user message content.') }}
                 {%- endif %}
             {%- endfor %}
             {{- '[/INST]' }}
@@ -72,7 +74,7 @@
             {{- raise_exception('User message must have a string or a list of chunks in content') }}
         {%- endif %}
 
-    {#- Assistant messages supports text content or text and thinking chunks. #}
+    {#- Assistant messages support text content or text and thinking chunks. #}
     {%- elif message['role'] == 'assistant' %}
         {%- if (message['content'] is none or message['content'] == '' or message['content']|length == 0) and (message['tool_calls'] is not defined or message['tool_calls'] is none or message['tool_calls']|length == 0) %}
             {{- raise_exception('Assistant message must have a string or a list of chunks in content or a list of tool calls.') }}
@@ -108,7 +110,7 @@
 
         {{- eos_token }}
 
-    {#- Tool messages only supports text content. #}
+    {#- Tool messages only support text content. #}
     {%- elif message['role'] == 'tool' %}
         {{- '[TOOL_RESULTS]' + message['content']|string + '[/TOOL_RESULTS]' }}
 
@@ -116,4 +118,4 @@
     {%- else %}
         {{- raise_exception('Only user, assistant and tool roles are supported, got ' + message['role'] + '.') }}
     {%- endif %}
-{%- endfor %}
+{%- endfor %}