Ver Fonte

fix: enhance memory cleaning condition to check model list length

myhloli há 4 meses atrás
pai
commit
f41fc406c3
1 ficheiros alterados com 1 adições e 1 exclusões
  1. 1 1
      mineru/backend/pipeline/model_json_to_middle_json.py

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

@@ -236,7 +236,7 @@ def result_to_middle_json(model_list, images_list, pdf_doc, image_writer, lang=N
 
     """清理内存"""
     pdf_doc.close()
-    if os.getenv('MINERU_DONOT_CLEAN_MEM') is None and len(model_list) > 10:
+    if os.getenv('MINERU_DONOT_CLEAN_MEM') is None and len(model_list) >= 10:
         clean_memory(get_device())
 
     return middle_json