Jelajahi Sumber

refactor(magic_pdf): improve code structure and memory safety

myhloli 9 bulan lalu
induk
melakukan
4021abeb98
1 mengubah file dengan 4 tambahan dan 3 penghapusan
  1. 4 3
      magic_pdf/model/doc_analyze_by_custom_model.py

+ 4 - 3
magic_pdf/model/doc_analyze_by_custom_model.py

@@ -1,16 +1,17 @@
 import os
 import os
 import time
 import time
 
 
+import torch
+
 # 关闭paddle的信号处理
 # 关闭paddle的信号处理
 import paddle
 import paddle
-import torch
+paddle.disable_signal_handler()
+
 from loguru import logger
 from loguru import logger
 
 
 from magic_pdf.model.batch_analyze import BatchAnalyze
 from magic_pdf.model.batch_analyze import BatchAnalyze
 from magic_pdf.model.sub_modules.model_utils import get_vram
 from magic_pdf.model.sub_modules.model_utils import get_vram
 
 
-paddle.disable_signal_handler()
-
 os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1'  # 禁止albumentations检查更新
 os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1'  # 禁止albumentations检查更新
 
 
 try:
 try: