[model] support granite4 (#8680)

2025-07-21 14:15:36 +08:00
parent 1639e4b587
commit 9d6565d1a8
4 changed files with 30 additions and 0 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -1029,6 +1029,17 @@ register_model_group(
 )


+register_model_group(
+    models={
+        "Granite-4.0-tiny-preview": {
+            DownloadSource.DEFAULT: "ibm-granite/granite-4.0-tiny-preview",
+            DownloadSource.MODELSCOPE: "ibm-granite/granite-4.0-tiny-preview",
+        },
+    },
+    template="granite4",
+)
+
+
 register_model_group(
    models={
        "Hunyuan-7B-Instruct": {