Explorar o código

fix: update MinerU version references in enum_class.py and header.html

myhloli hai 2 meses
pai
achega
6ea2cfeb21
Modificáronse 2 ficheiros con 3 adicións e 3 borrados
  1. 1 1
      mineru/resources/header.html
  2. 2 2
      mineru/utils/enum_class.py

+ 1 - 1
mineru/resources/header.html

@@ -54,7 +54,7 @@
             font-family: 'Trebuchet MS', 'Lucida Sans Unicode',
               'Lucida Grande', 'Lucida Sans', Arial, sans-serif;
           ">
-          MinerU 2.5: PDF Extraction Demo
+          MinerU 2: PDF Extraction Demo
         </h1>
       </div>
     </div>

+ 2 - 2
mineru/utils/enum_class.py

@@ -63,8 +63,8 @@ class MakeMode:
 
 
 class ModelPath:
-    vlm_root_hf = "opendatalab/MinerU2.0-2505-0.9B"
-    vlm_root_modelscope = "OpenDataLab/MinerU2.0-2505-0.9B"
+    vlm_root_hf = "opendatalab/MinerU2.5-2509-1.2B"
+    vlm_root_modelscope = "OpenDataLab/MinerU2.5-2509-1.2B"
     pipeline_root_modelscope = "OpenDataLab/PDF-Extract-Kit-1.0"
     pipeline_root_hf = "opendatalab/PDF-Extract-Kit-1.0"
     doclayout_yolo = "models/Layout/YOLO/doclayout_yolo_docstructbench_imgsz1280_2501.pt"