Parcourir la source

fix: conditionally clean memory based on environment variable

myhloli il y a 4 mois
Parent
commit
12b64a0e46
1 fichiers modifiés avec 3 ajouts et 1 suppressions
  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