Spaces:
Running
Running
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"
|