jhj0517 commited on
Commit
342afe7
·
1 Parent(s): 9212c90

fix spaces bug

Browse files
modules/whisper/faster_whisper_inference.py CHANGED
@@ -144,8 +144,8 @@ class FasterWhisperInference(WhisperBase):
144
  @spaces.GPU
145
  def get_available_compute_type(self):
146
  if self.device == "cuda":
147
- return ctranslate2.get_supported_compute_types("cuda")
148
- return ctranslate2.get_supported_compute_types("cpu")
149
 
150
  @staticmethod
151
  def get_device():
 
144
  @spaces.GPU
145
  def get_available_compute_type(self):
146
  if self.device == "cuda":
147
+ return ['float32', 'int8_float16', 'float16', 'int8', 'int8_float32']
148
+ return ['int16', 'float32', 'int8', 'int8_float32']
149
 
150
  @staticmethod
151
  def get_device():