Browse Source

Merge pull request #2836 from myhloli/dev

fix: conditionally clean memory based on environment variable
Xiaomeng Zhao 4 months ago
parent
commit
23bb358736
1 changed files with 3 additions and 1 deletions
  1. 3 1
      mineru/backend/pipeline/model_json_to_middle_json.py

+ 3 - 1
mineru/backend/pipeline/model_json_to_middle_json.py

@@ -1,4 +1,5 @@
 # Copyright (c) Opendatalab. All rights reserved.
+import os
 import time
 
 from loguru import logger
@@ -235,7 +236,8 @@ def result_to_middle_json(model_list, images_list, pdf_doc, image_writer, lang=N
 
     """清理内存"""
     pdf_doc.close()
-    clean_memory(get_device())
+    if os.getenv('MINERU_DONOT_CLEAN_MEM') is None:
+        clean_memory(get_device())
 
     return middle_json