Skip to content

GitLab

Unverified Commit 23bb3587 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Merge pull request #2836 from myhloli/dev 

fix: conditionally clean memory based on environment variable
parents d39aa87e 12b64a0e
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
mineru/backend/pipeline/model_json_to_middle_json.py
View file @ 23bb3587
# Copyright (c) Opendatalab. All rights reserved. # Copyright (c) Opendatalab. All rights reserved.
import os
import time import time
from loguru import logger from loguru import logger
...@@ -235,6 +236,7 @@ def result_to_middle_json(model_list, images_list, pdf_doc, image_writer, lang=N ...@@ -235,6 +236,7 @@ def result_to_middle_json(model_list, images_list, pdf_doc, image_writer, lang=N
"""清理内存""" """清理内存"""
pdf_doc.close() pdf_doc.close()
if os.getenv('MINERU_DONOT_CLEAN_MEM') is None:
clean_memory(get_device()) clean_memory(get_device())
return middle_json return middle_json
......
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