From 88b909e20e0e27da406e694bf0010ea229e54406 Mon Sep 17 00:00:00 2001
From: myhloli <moe@myhloli.com>
Date: Mon, 30 Dec 2024 14:18:02 +0800
Subject: [PATCH] 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
---
 magic_pdf/model/sub_modules/model_init.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/magic_pdf/model/sub_modules/model_init.py b/magic_pdf/model/sub_modules/model_init.py
index d5ff7ee0..0eeed246 100644
--- a/magic_pdf/model/sub_modules/model_init.py
+++ b/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,
                    ):
 
-    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