[data] fix gemma2 eos token (#8480)

Co-authored-by: Yaowei Zheng <hiyouga@buaa.edu.cn>
2025-06-27 18:19:15 +08:00
parent 2c26ce6ac4
commit 4f0da0aec9
6 changed files with 40 additions and 4 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -712,6 +712,13 @@ register_model_group(
        "Gemma-1.1-7B-Instruct": {
            DownloadSource.DEFAULT: "google/gemma-1.1-7b-it",
        },
+    },
+    template="gemma",
+)
+
+
+register_model_group(
+    models={
        "Gemma-2-2B": {
            DownloadSource.DEFAULT: "google/gemma-2-2b",
            DownloadSource.MODELSCOPE: "LLM-Research/gemma-2-2b",
@@ -751,7 +758,7 @@ register_model_group(
            DownloadSource.MODELSCOPE: "google/medgemma-27b-text-it",
        },
    },
-    template="gemma",
+    template="gemma2",
 )