Update modeling_openpangu_dense.py

modeling_openpangu_dense.py

@@ -31,12 +31,13 @@ from typing import Callable, Optional, Union
 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.activations import ACT2FN