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 "cpu"
 
145
  def get_device():
146
  if torch.cuda.is_available():
147
  return "cuda"
 
 
148
  else:
149
+ return "auto"