jhj0517 commited on
Commit
24956db
·
1 Parent(s): d79b9a9

fix spaces bug

Browse files
modules/whisper/faster_whisper_inference.py CHANGED
@@ -29,8 +29,7 @@ class FasterWhisperInference(WhisperBase):
29
  self.model_paths = self.get_model_paths()
30
  self.device = self.get_device()
31
  self.available_models = self.model_paths.keys()
32
- self.available_compute_types = ctranslate2.get_supported_compute_types(
33
- "cuda") if self.device == "cuda" else ctranslate2.get_supported_compute_types("cpu")
34
 
35
  def transcribe(self,
36
  audio: Union[str, BinaryIO, np.ndarray],
 
29
  self.model_paths = self.get_model_paths()
30
  self.device = self.get_device()
31
  self.available_models = self.model_paths.keys()
32
+ self.available_compute_types = self.get_available_compute_type()
 
33
 
34
  def transcribe(self,
35
  audio: Union[str, BinaryIO, np.ndarray],