Spaces:
Running
Running
jhj0517
commited on
Commit
·
24956db
1
Parent(s):
d79b9a9
fix spaces bug
Browse files
modules/whisper/faster_whisper_inference.py
CHANGED
@@ -29,8 +29,7 @@ class FasterWhisperInference(WhisperBase):
|
|
29 |
self.model_paths = self.get_model_paths()
|
30 |
self.device = self.get_device()
|
31 |
self.available_models = self.model_paths.keys()
|
32 |
-
self.available_compute_types =
|
33 |
-
"cuda") if self.device == "cuda" else ctranslate2.get_supported_compute_types("cpu")
|
34 |
|
35 |
def transcribe(self,
|
36 |
audio: Union[str, BinaryIO, np.ndarray],
|
|
|
29 |
self.model_paths = self.get_model_paths()
|
30 |
self.device = self.get_device()
|
31 |
self.available_models = self.model_paths.keys()
|
32 |
+
self.available_compute_types = self.get_available_compute_type()
|
|
|
33 |
|
34 |
def transcribe(self,
|
35 |
audio: Union[str, BinaryIO, np.ndarray],
|