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"
|