magic_pdf/model/pdf_extract_kit.py · 1c9f9942d8d91edc07a3bc0b06e965a9929b8b67 · wangsen / MinerU

fix(device): enable MPS support and fix related issues · 203b8f90

myhloli authored Jan 10, 2025

- Add MPS support for Apple Silicon devices
- Implement empty_cache() for MPS devices
- Set PYTORCH_ENABLE_MPS_FALLBACK environment variable
- Adjust MFR model device allocation for MPS

203b8f90

pdf_extract_kit.py 12 KB

Replace pdf_extract_kit.py