fix bug

modules/whisper/whisper_base.py

@@ -399,15 +399,15 @@ class WhisperBase(ABC):
 
         return time_str.strip()
 
-    # @staticmethod
-    # @spaces.GPU(duration=120)
-    # def get_device():
-    #     if torch.cuda.is_available():
-    #         return "cuda"
-    #     elif torch.backends.mps.is_available():
-    #         return "mps"
-    #     else:
-    #         return "cpu"
+    @staticmethod
+    @spaces.GPU(duration=120)
+    def get_device():
+        if torch.cuda.is_available():
+            return "cuda"
+        elif torch.backends.mps.is_available():
+            return "mps"
+        else:
+            return "cpu"
 
     @staticmethod
     @spaces.GPU(duration=120)