jhj0517 commited on
Commit
b39eb49
·
1 Parent(s): 89f23c0
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +2 -2
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,