add codegemma

Former-commit-id: 6543f3d449
2026-03-08 12:46:06 +08:00 · 2024-04-16 00:11:15 +08:00
parent 2dc3343b1c
commit bd2b758b48
4 changed files with 23 additions and 7 deletions
--- a/src/llmtuner/extras/constants.py
+++ b/src/llmtuner/extras/constants.py
@@ -385,6 +385,23 @@ register_model_group(
 )


+register_model_group(
+    models={
+        "CodeGemma-2B": {
+            DownloadSource.DEFAULT: "google/codegemma-2b",
+        },
+        "CodeGemma-7B": {
+            DownloadSource.DEFAULT: "google/codegemma-7b",
+        },
+        "CodeGemma-7B-Chat": {
+            DownloadSource.DEFAULT: "google/codegemma-7b-it",
+            DownloadSource.MODELSCOPE: "AI-ModelScope/codegemma-7b-it",
+        },
+    },
+    template="gemma",
+)
+
+
 register_model_group(
    models={
        "InternLM-7B": {