Browse Source

fix: remove commented-out code for autocasting in prediction logic of predict_formula.py

myhloli 3 tuần trước cách đây
mục cha
commit
5ade203e31
2 tập tin đã thay đổi với 3 bổ sung3 xóa
  1. 2 2
      mineru/backend/pipeline/model_init.py
  2. 1 1
      mineru/cli/models_download.py

+ 2 - 2
mineru/backend/pipeline/model_init.py

@@ -19,8 +19,8 @@ from ...utils.models_download_utils import auto_download_and_get_model_root_path
 
 MFR_MODEL = os.getenv('MINERU_MFR_MODEL', None)
 if MFR_MODEL is None:
-    # MFR_MODEL = "unimernet_small"
-    MFR_MODEL = "pp_formulanet_plus_m"
+    MFR_MODEL = "unimernet_small"
+    # MFR_MODEL = "pp_formulanet_plus_m"
 
 
 def img_orientation_cls_model_init():

+ 1 - 1
mineru/cli/models_download.py

@@ -63,7 +63,7 @@ def download_pipeline_models():
     model_paths = [
         ModelPath.doclayout_yolo,
         ModelPath.yolo_v8_mfd,
-        # ModelPath.unimernet_small,
+        ModelPath.unimernet_small,
         ModelPath.pytorch_paddle,
         ModelPath.layout_reader,
         ModelPath.slanet_plus,