From ad205ad02a5196c888e186cc85c6b56cdc41bd33 Mon Sep 17 00:00:00 2001
From: 51616 <rujikorn.ch@gmail.com>
Date: Tue, 10 Jun 2025 03:08:16 +0000
Subject: [PATCH] fix chat template (include chat eos in labels)

---
 chat_templates/google/gemma-2-2b-it.jinja | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/chat_templates/google/gemma-2-2b-it.jinja b/chat_templates/google/gemma-2-2b-it.jinja
index c29c4e2..23129b0 100644
--- a/chat_templates/google/gemma-2-2b-it.jinja
+++ b/chat_templates/google/gemma-2-2b-it.jinja
@@ -14,7 +14,7 @@
     {%- if message['role'] == 'user' and loop.first and system_message is defined %}
         {{ '<start_of_turn>' + role + '\n' + system_message + '\n\n' + message['content'] | trim + '<end_of_turn>\n' }}
     {%- elif message['role'] == 'assistant' %}
-        {{ '<start_of_turn>' + role + '\n' }}{% generation %}{{ message['content'] | trim }}{% endgeneration %}{{ '<end_of_turn>\n' }}
+        {{ '<start_of_turn>' + role + '\n' }}{% generation %}{{ message['content'] | trim + '<end_of_turn>\n' }}{% endgeneration %}
     {%- else %}
         {{ '<start_of_turn>' + role + '\n' + message['content'] | trim + '<end_of_turn>\n' }}
     {%- endif %}