"vscode:/vscode.git/clone" did not exist on "dfcd459e5f0e3906760712c4a5f507dd846ee518"
Commit 4021abeb authored by myhloli's avatar myhloli
Browse files

refactor(magic_pdf): improve code structure and memory safety

parent 4c0af020
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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment