jhj0517 commited on
Commit
87cbb02
·
1 Parent(s): b39eb49

Use float32 for cpu

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +2 -0
modules/whisper/whisper_base.py CHANGED
@@ -375,6 +375,8 @@ class WhisperBase(ABC):
375
  def get_compute_type(self):
376
  if "float16" in self.available_compute_types:
377
  return "float16"
 
 
378
  else:
379
  return self.available_compute_types[0]
380
 
 
375
  def get_compute_type(self):
376
  if "float16" in self.available_compute_types:
377
  return "float16"
378
+ if "float32" in self.available_compute_types:
379
+ return "float32"
380
  else:
381
  return self.available_compute_types[0]
382