magic_pdf/model/doc_analyze_by_custom_model.py · a2cde43b57746a8989a9cd09728dbed753ffebfd · wangsen / MinerU

fix(model): improve VRAM detection and handling · d32a63ca

myhloli authored Apr 07, 2025

- Refactor VRAM detection logic for better readability and efficiency
- Add fallback mechanism for unknown VRAM sizes
- Improve device checking in get_vram function

d32a63ca

doc_analyze_by_custom_model.py 10 KB

Replace doc_analyze_by_custom_model.py