jhj0517 commited on
Commit
81956d0
·
unverified ·
2 Parent(s): 144a711 0531958

Merge pull request #194 from jhj0517/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"