From aabed4edc62fe97bdaa0a04c8d355660378d27e9 Mon Sep 17 00:00:00 2001 From: myhloli Date: Fri, 30 Jan 2026 21:27:07 +0800 Subject: [PATCH] feat: add MLU support for compute capability detection in utils.py --- mineru/backend/vlm/utils.py | 2 ++ 1 file changed, 2 insertions(+) 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