Fix device error

modules/whisper/whisper_base.py

@@ -453,15 +453,34 @@ class WhisperBase(ABC):
 
         return time_str.strip()
 
-    @staticmethod
-    def get_device():
+    def get_device(self):
         if torch.cuda.is_available():
             return "cuda"
         elif torch.backends.mps.is_available():
+            if self.is_sparse_mps_supported():
+                # MPS is not supported for sparse tensor for now. See : https://github.com/pytorch/pytorch/issues/87886
+                return "cpu"
             return "mps"
         else:
             return "cpu"
 
+    @staticmethod
+    def is_sparse_mps_supported():
+        if torch.backends.mps.is_available():
+            return False
+
+        try:
+            device = torch.device("mps")
+            sparse_tensor = torch.sparse_coo_tensor(
+                indices=torch.tensor([[0, 1], [2, 3]]),
+                values=torch.tensor([1, 2]),
+                size=(4, 4),
+                device=device
+            )
+            return True
+        except RuntimeError:
+            return False
+
     @staticmethod
     def release_cuda_memory():
         """Release memory"""