diff --git a/mineru/backend/vlm/utils.py b/mineru/backend/vlm/utils.py index a5aac6e3..24b04b90 100644 --- a/mineru/backend/vlm/utils.py +++ b/mineru/backend/vlm/utils.py @@ -22,6 +22,8 @@ def enable_custom_logits_processors() -> bool: compute_capability = "8.0" elif hasattr(torch, 'musa') and torch.musa.is_available(): compute_capability = "8.0" + elif hasattr(torch, 'mlu') and torch.mlu.is_available(): + compute_capability = "8.0" else: logger.info("CUDA not available, disabling custom_logits_processors") return False