opensource/MinerU
3
0
Fork 0
mirror of https://github.com/opendatalab/MinerU.git synced 2026-09-24 23:10:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(magic_pdf): comment out npu-related code

Browse Source 
- Remove use_npu variable initialization
- Comment out device assignment and npu check
- Comment out use_npu parameter in ModifiedPaddleOCR constructor
This commit is contained in:
myhloli
2024-12-30 14:18:02 +08:00
parent 2684e7753b
commit 88b909e20e
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
magic_pdf/model/sub_modules/model_init.py
+6 -6
View File
@@ -70,10 +70,10 @@ def ocr_model_init(show_log: bool = False,
det_db_unclip_ratio=1.8,
):
use_npu = False
device = get_device()
if str(device).startswith("npu"):
use_npu = True
# use_npu = False
# device = get_device()
# if str(device).startswith("npu"):
# use_npu = True
if lang is not None and lang != '':
model = ModifiedPaddleOCR(
@@ -82,7 +82,7 @@ def ocr_model_init(show_log: bool = False,
lang=lang,
use_dilation=use_dilation,
det_db_unclip_ratio=det_db_unclip_ratio,
use_npu=use_npu,
# use_npu=use_npu,
)
else:
model = ModifiedPaddleOCR(
@@ -90,7 +90,7 @@ def ocr_model_init(show_log: bool = False,
det_db_box_thresh=det_db_box_thresh,
use_dilation=use_dilation,
det_db_unclip_ratio=det_db_unclip_ratio,
use_npu=use_npu,
# use_npu=use_npu,
)
return model