From 13a062fad472c3c5685d64cb4ea6ff1686030165 Mon Sep 17 00:00:00 2001
From: 51616 <rujikorn.ch@gmail.com>
Date: Wed, 25 Jun 2025 23:20:53 +0900
Subject: [PATCH] fix template for gemm3-1b

---
 chat_templates/google/gemma-3-1b-it.jinja | 35 +++++------------------
 1 file changed, 7 insertions(+), 28 deletions(-)

diff --git a/chat_templates/google/gemma-3-1b-it.jinja b/chat_templates/google/gemma-3-1b-it.jinja
index 2bfe311..d4e3fe0 100644
--- a/chat_templates/google/gemma-3-1b-it.jinja
+++ b/chat_templates/google/gemma-3-1b-it.jinja
@@ -1,13 +1,9 @@
 {{ bos_token }}
 {%- if messages[0]['role'] == 'system' -%}
     {%- if messages[0]['content'] is string -%}
-        {%- set first_user_prefix = messages[0]['content'] + '
-
-' -%}
+        {%- set first_user_prefix = messages[0]['content'] + '\n\n' -%}
     {%- else -%}
-        {%- set first_user_prefix = messages[0]['content'][0]['text'] + '
-
-' -%}
+        {%- set first_user_prefix = messages[0]['content'][0]['text'] + '\n\n' -%}
     {%- endif -%}
     {%- set loop_messages = messages[1:] -%}
 {%- else -%}
@@ -23,29 +19,12 @@
     {%- else -%}
         {%- set role = message['role'] -%}
     {%- endif -%}
-    {{ '<start_of_turn>' + role + '
-' + (first_user_prefix if loop.first else "") }}
-    {%- if message['content'] is string -%}
-        {%- if message['role'] == 'assistant' -%}
-            {% generation %}{{ message['content'] | trim + '<end_of_turn>
-' }}{% endgeneration %}
-        {%- else -%}
-            {{ message['content'] | trim + '<end_of_turn>
-' }}
-        {%- endif -%}
-    {# {%- elif message['content'] is iterable -%}
-        {%- for item in message['content'] -%}
-            {%- if item['type'] == 'image' -%}
-                {{ '<start_of_image>' }}
-            {%- elif item['type'] == 'text' -%}
-                {{ item['text'] | trim }}
-            {%- endif -%}
-        {%- endfor -%} #}
-    {%- else -%}
-        {{ raise_exception("Invalid content type") }}
+    {%- if (message['role'] == 'user') -%}
+        {{ '<start_of_turn>' + role + '\n' + message['content'] | trim + '<end_of_turn>\n' }}
+    {%- elif (message['role'] == 'assistant') -%}
+        {{ '<start_of_turn>' + role + '\n' }}{% generation %}{{ message['content'] | trim + '<end_of_turn>\n' }}{% endgeneration %}
     {%- endif -%}
 {%- endfor -%}
 {%- if add_generation_prompt -%}
-    {{'<start_of_turn>model
-'}}
+    {{'<start_of_turn>model\n'}}
 {%- endif -%}
\ No newline at end of file