[model] support GLM-OCR SFT (#10183)

2026-03-02 01:36:02 +08:00 · 2026-02-10 21:41:01 +08:00
parent 1d5e8ebcd0
commit d3ebd5678d
4 changed files with 38 additions and 0 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -950,6 +950,18 @@ register_model_group(
 )


+register_model_group(
+    models={
+        "GLM-OCR": {
+            DownloadSource.DEFAULT: "zai-org/GLM-OCR",
+            DownloadSource.MODELSCOPE: "ZhipuAI/GLM-OCR",
+        },
+    },
+    template="glm_ocr",
+    multimodal=True,
+)
+
+
 register_model_group(
    models={
        "GLM-Z1-0414-9B-Chat": {