Эх сурвалжийг харах

refactor(magic_pdf): comment out npu-related code

- Remove use_npu variable initialization
- Comment out device assignment and npu check
- Comment out use_npu parameter in ModifiedPaddleOCR constructor
myhloli 10 сар өмнө
parent
commit
88b909e20e

+ 6 - 6
magic_pdf/model/sub_modules/model_init.py

@@ -70,10 +70,10 @@ def ocr_model_init(show_log: bool = False,
                    det_db_unclip_ratio=1.8,
                    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 != '':
     if lang is not None and lang != '':
         model = ModifiedPaddleOCR(
         model = ModifiedPaddleOCR(
@@ -82,7 +82,7 @@ def ocr_model_init(show_log: bool = False,
             lang=lang,
             lang=lang,
             use_dilation=use_dilation,
             use_dilation=use_dilation,
             det_db_unclip_ratio=det_db_unclip_ratio,
             det_db_unclip_ratio=det_db_unclip_ratio,
-            use_npu=use_npu,
+            # use_npu=use_npu,
         )
         )
     else:
     else:
         model = ModifiedPaddleOCR(
         model = ModifiedPaddleOCR(
@@ -90,7 +90,7 @@ def ocr_model_init(show_log: bool = False,
             det_db_box_thresh=det_db_box_thresh,
             det_db_box_thresh=det_db_box_thresh,
             use_dilation=use_dilation,
             use_dilation=use_dilation,
             det_db_unclip_ratio=det_db_unclip_ratio,
             det_db_unclip_ratio=det_db_unclip_ratio,
-            use_npu=use_npu,
+            # use_npu=use_npu,
         )
         )
     return model
     return model