Spaces:
Running
Running
jhj0517
commited on
Commit
·
b39eb49
1
Parent(s):
89f23c0
Fix type
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -380,9 +380,9 @@ class WhisperBase(ABC):
|
|
380 |
|
381 |
def get_available_compute_type(self):
|
382 |
if self.device == "cuda":
|
383 |
-
return ctranslate2.get_supported_compute_types("cuda")
|
384 |
else:
|
385 |
-
return ctranslate2.get_supported_compute_types("cpu")
|
386 |
|
387 |
@staticmethod
|
388 |
def generate_and_write_file(file_name: str,
|
|
|
380 |
|
381 |
def get_available_compute_type(self):
|
382 |
if self.device == "cuda":
|
383 |
+
return list(ctranslate2.get_supported_compute_types("cuda"))
|
384 |
else:
|
385 |
+
return list(ctranslate2.get_supported_compute_types("cpu"))
|
386 |
|
387 |
@staticmethod
|
388 |
def generate_and_write_file(file_name: str,
|