Эх сурвалжийг харах

Update mineru/backend/vlm/utils.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Xiaomeng Zhao 1 долоо хоног өмнө
parent
commit
ab365420b9

+ 1 - 1
mineru/backend/vlm/utils.py

@@ -16,7 +16,7 @@ def enable_custom_logits_processors() -> bool:
         major, minor = torch.cuda.get_device_capability()
         # 正确计算Compute Capability
         compute_capability = f"{major}.{minor}"
-    elif torch.npu.is_available():
+    elif hasattr(torch, 'npu') and torch.npu.is_available():
         compute_capability = "8.0"
     else:
         logger.info("CUDA not available, disabling custom_logits_processors")