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):
|