[model] add dots ocr (#9176)

2025-09-21 23:34:19 +08:00
parent 800934b507
commit 80fe3a172d
4 changed files with 45 additions and 3 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -601,6 +601,18 @@ register_model_group(
 )


+register_model_group(
+    models={
+        "dots.ocr": {
+            DownloadSource.DEFAULT: "rednote-hilab/dots.ocr",
+            DownloadSource.MODELSCOPE: "rednote-hilab/dots.ocr",
+        },
+    },
+    template="dots_ocr",
+    multimodal=True,
+)
+
+
 register_model_group(
    models={
        "ERNIE-4.5-21B-A3B-Thinking": {