Update modular_openpangu_dense.py

modular_openpangu_dense.py

@@ -24,12 +24,13 @@ from typing import Callable, Optional, Tuple
 import torch
 from torch import nn
 
-import torch_npu
-from torch_npu.contrib import transfer_to_npu
-if "910" in torch.npu.get_device_name():
-    NPU_ATTN_INFR = True
-    print("[INFO] torch_npu detected. Using NPU fused infer attention.")
-else:
+try:
+    import torch_npu
+    from torch_npu.contrib import transfer_to_npu
+    if "910" in torch.npu.get_device_name():
+        NPU_ATTN_INFR = True
+        print("[INFO] torch_npu detected. Using NPU fused infer attention.")
+except ImportError:
     NPU_ATTN_INFR = False
 
 from transformers.cache_utils import Cache