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

refactor(magic_pdf): improve code structure and memory safety

myhloli 9 сар өмнө
parent
commit
4021abeb98

+ 4 - 3
magic_pdf/model/doc_analyze_by_custom_model.py

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