Spaces:
Running
Running
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 |
-
#
|
159 |
-
return "
|
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):
|