magic_pdf/model/sub_modules/model_utils.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

model_utils.py 2.59 KB

Replace model_utils.py