parallel fixes

marker/convert.py

@@ -96,7 +96,6 @@ def convert_single_pdf(
 
     # OCR pages as needed
     pages, ocr_stats = run_ocr(doc, pages, langs, ocr_model, batch_multiplier=batch_multiplier, ocr_all_pages=ocr_all_pages)
-    flush_cuda_memory()
 
     out_meta["ocr_stats"] = ocr_stats
     if len([b for p in pages for b in p.blocks]) == 0:

marker_app.py

@@ -1,6 +1,7 @@
 import os
 os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
 os.environ["IN_STREAMLIT"] = "true"
+os.environ["PDFTEXT_CPU_WORKERS"] = "1"
 
 import base64
 import io

run_marker_app.py

@@ -7,7 +7,7 @@ def run_app():
     cur_dir = os.path.dirname(os.path.abspath(__file__))
     app_path = os.path.join(cur_dir, "marker_app.py")
     cmd = ["streamlit", "run", app_path]
-    subprocess.run(cmd, env={**os.environ, "IN_STREAMLIT": "true"})
+    subprocess.run(cmd, env={**os.environ, "IN_STREAMLIT": "true", "PDFTEXT_CPU_WORKERS": "1"})
 
 
 if __name__ == "__main__":