瀏覽代碼

fix: revert MFR_MODEL to unimernet_small in model initialization

myhloli 4 周之前
父節點
當前提交
cd0ed982c0
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      mineru/backend/pipeline/model_init.py

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

@@ -17,8 +17,8 @@ from ...model.table.rec.unet_table.main import UnetTableModel
 from ...utils.enum_class import ModelPath
 from ...utils.enum_class import ModelPath
 from ...utils.models_download_utils import auto_download_and_get_model_root_path
 from ...utils.models_download_utils import auto_download_and_get_model_root_path
 
 
-# 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():
 def img_orientation_cls_model_init():