Skip to content

GitLab

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

refactor(magic_pdf): improve code structure and memory safety

parent 4c0af020
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
magic_pdf/model/doc_analyze_by_custom_model.py
View file @ 4021abeb
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:
......
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