add check hasattr(torch, "musa")

Author: hanhainebula

FlagEmbedding/abc/inference/AbsReranker.py

@@ -112,7 +112,7 @@ def get_target_devices(devices: Union[str, int, List[str], List[int]]) -> List[s
                 return [f"cuda:{i}" for i in range(torch.cuda.device_count())]
             elif is_torch_npu_available():
                 return [f"npu:{i}" for i in range(torch.npu.device_count())]
-            elif torch.musa.is_available():
+            elif hasattr(torch, "musa") and torch.musa.is_available():
                 return [f"musa:{i}" for i in range(torch.musa.device_count())]
             elif torch.backends.mps.is_available():
                 return ["mps"]
@@ -121,15 +121,15 @@ def get_target_devices(devices: Union[str, int, List[str], List[int]]) -> List[s
         elif isinstance(devices, str):
             return [devices]
         elif isinstance(devices, int):
-            if torch.musa.is_available():
+            if hasattr(torch, "musa") and torch.musa.is_available():
                 return [f"musa:{devices}"]
             else:
                 return [f"cuda:{devices}"]
         elif isinstance(devices, list):
             if isinstance(devices[0], str):
                 return devices
             elif isinstance(devices[0], int):
-                if torch.musa.is_available():
+                if hasattr(torch, "musa") and torch.musa.is_available():
                     return [f"musa:{device}" for device in devices]
                 else:
                     return [f"cuda:{device}" for device in devices]