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,
|