Преглед на файлове

fix(magic_pdf): disable PaddlePaddle signal handler

- Import paddle module and disable its signal handler to prevent interference with other components
- This change addresses potential conflicts between PaddlePaddle and other libraries or system signals
myhloli преди 11 месеца
родител
ревизия
dd7f6781f6
променени са 1 файла, в които са добавени 4 реда и са изтрити 0 реда
  1. 4 0
      magic_pdf/model/doc_analyze_by_custom_model.py

+ 4 - 0
magic_pdf/model/doc_analyze_by_custom_model.py

@@ -5,6 +5,10 @@ import fitz
 import numpy as np
 from loguru import logger
 
+# 关闭paddle的信号处理
+import paddle
+paddle.disable_signal_handler()
+
 os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1'  # 禁止albumentations检查更新
 os.environ['YOLO_VERBOSE'] = 'False'  # disable yolo logger