Преглед на файлове

fix: conditionally clean memory based on environment variable

myhloli преди 4 месеца
родител
ревизия
12b64a0e46
променени са 1 файла, в които са добавени 3 реда и са изтрити 1 реда
  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