Spaces:
Runtime error
Runtime error
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,
|