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