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

2026-05-28 10:58:54 +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/data/collator.py
+++ b/src/llamafactory/data/collator.py
@@ -213,6 +213,7 @@ class MultiModalDataCollatorForSeq2Seq(DataCollatorForSeq2Seq):
            and getattr(self.model.config, "model_type", None)
            in [
                "glm4v",
+                "glm_ocr",
                "Keye",
                "qwen2_vl",
                "qwen2_5_vl",