magic_pdf/model/doc_analyze_by_custom_model.py · 052a4d72ed37ceba16ec6ac9aa6785edab7b11b1 · wangsen / MinerU

perf(magic_pdf): optimize batch ratio calculation for GPU · 052a4d72

myhloli authored Jan 21, 2025

- Update GPU memory check and batch ratio calculation logic
- Add support for virtual VRAM size environment variable
- Improve logging for GPU memory and batch ratio

052a4d72

doc_analyze_by_custom_model.py 7.67 KB

Replace doc_analyze_by_custom_model.py