Merge pull request #1591 from myhloli/dev

fix(models): update unimernet_small model path
opendatalab · Jan 21, 2025 · 6be3277 · 6be3277
2 parents 58d28c9 + 2a3a006
commit 6be3277
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/magic_pdf/resources/model_config/model_configs.yaml b/magic_pdf/resources/model_config/model_configs.yaml
@@ -2,7 +2,7 @@ weights:
   layoutlmv3: Layout/LayoutLMv3/model_final.pth
   doclayout_yolo: Layout/YOLO/doclayout_yolo_docstructbench_imgsz1280_2501.pt
   yolo_v8_mfd: MFD/YOLO/yolo_v8_ft.pt
-  unimernet_small: MFR/unimernet_small
+  unimernet_small: MFR/unimernet_small_2501
   struct_eqtable: TabRec/StructEqTable
   tablemaster: TabRec/TableMaster
   rapid_table: TabRec/RapidTable
diff --git a/projects/gradio_app/examples/complex_layout.pdf b/projects/gradio_app/examples/complex_layout.pdf
diff --git a/scripts/download_models.py b/scripts/download_models.py
@@ -35,7 +35,7 @@ def download_and_modify_json(url, local_filename, modifications):
         "models/Layout/LayoutLMv3/*",
         "models/Layout/YOLO/*",
         "models/MFD/YOLO/*",
-        "models/MFR/unimernet_small/*",
+        "models/MFR/unimernet_small_2501/*",
         "models/TabRec/TableMaster/*",
         "models/TabRec/StructEqTable/*",
     ]

diff --git a/scripts/download_models_hf.py b/scripts/download_models_hf.py
@@ -36,7 +36,7 @@ def download_and_modify_json(url, local_filename, modifications):
         "models/Layout/LayoutLMv3/*",
         "models/Layout/YOLO/*",
         "models/MFD/YOLO/*",
-        "models/MFR/unimernet_small/*",
+        "models/MFR/unimernet_small_2501/*",
         "models/TabRec/TableMaster/*",
         "models/TabRec/StructEqTable/*",
     ]