ソースを参照

Update mineru/backend/vlm/utils.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Xiaomeng Zhao 1 週間 前
コミット
ab365420b9
1 ファイル変更1 行追加1 行削除
  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")