Explorar o código

Update mineru/backend/vlm/utils.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Xiaomeng Zhao hai 1 semana
pai
achega
ab365420b9
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      mineru/backend/vlm/utils.py

+ 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")