浏览代码

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
 import numpy as np
 from loguru import logger
 from loguru import logger
 
 
+# 关闭paddle的信号处理
+import paddle
+paddle.disable_signal_handler()
+
 os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1'  # 禁止albumentations检查更新
 os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1'  # 禁止albumentations检查更新
 os.environ['YOLO_VERBOSE'] = 'False'  # disable yolo logger
 os.environ['YOLO_VERBOSE'] = 'False'  # disable yolo logger