Spaces:
Runtime error
Runtime error
jhj0517 commited on
Commit ·
3a57221
1
Parent(s): 5cc743b
Fix spaces bug
Browse files
modules/whisper/faster_whisper_inference.py
CHANGED
|
@@ -151,9 +151,8 @@ class FasterWhisperInference(WhisperBase):
|
|
| 151 |
return ['int16', 'float32', 'int8', 'int8_float32']
|
| 152 |
|
| 153 |
def get_device(self):
|
| 154 |
-
|
| 155 |
-
|
| 156 |
-
return "cpu"
|
| 157 |
|
| 158 |
@staticmethod
|
| 159 |
def download_model(model_size: str, model_dir: str):
|
|
|
|
| 151 |
return ['int16', 'float32', 'int8', 'int8_float32']
|
| 152 |
|
| 153 |
def get_device(self):
|
| 154 |
+
# To fix huggingface spaces bug just return cuda
|
| 155 |
+
return "cuda"
|
|
|
|
| 156 |
|
| 157 |
@staticmethod
|
| 158 |
def download_model(model_size: str, model_dir: str):
|