jhj0517 commited on
Commit
b3a5df2
·
1 Parent(s): 8b2db3d

debug decoration

Browse files
modules/whisper/faster_whisper_inference.py CHANGED
@@ -153,10 +153,9 @@ class FasterWhisperInference(WhisperBase):
153
  return ['float32', 'int8_float16', 'float16', 'int8', 'int8_float32']
154
  return ['int16', 'float32', 'int8', 'int8_float32']
155
 
156
- @spaces.GPU
157
  def get_device(self):
158
- # To fix huggingface spaces bug just return cuda
159
- return "cuda"
160
 
161
  @staticmethod
162
  def download_model(model_size: str, model_dir: str):
 
153
  return ['float32', 'int8_float16', 'float16', 'int8', 'int8_float32']
154
  return ['int16', 'float32', 'int8', 'int8_float32']
155
 
 
156
  def get_device(self):
157
+ # Because of huggingface spaces bug, just return cpu
158
+ return "cpu"
159
 
160
  @staticmethod
161
  def download_model(model_size: str, model_dir: str):