소스 검색

fix: enhance memory cleaning condition to check model list length

myhloli 4 달 전
부모
커밋
f41fc406c3
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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