Spaces:
Runtime error
Runtime error
jhj0517 commited on
Commit ·
0531958
1
Parent(s): 144a711
Fix `get_device()``
Browse files
modules/whisper/faster_whisper_inference.py
CHANGED
|
@@ -145,7 +145,5 @@ class FasterWhisperInference(WhisperBase):
|
|
| 145 |
def get_device():
|
| 146 |
if torch.cuda.is_available():
|
| 147 |
return "cuda"
|
| 148 |
-
elif torch.backends.mps.is_available():
|
| 149 |
-
return "auto"
|
| 150 |
else:
|
| 151 |
-
return "
|
|
|
|
| 145 |
def get_device():
|
| 146 |
if torch.cuda.is_available():
|
| 147 |
return "cuda"
|
|
|
|
|
|
|
| 148 |
else:
|
| 149 |
+
return "auto"
|