Merge pull request #50 from jhj0517/patch#1

modules/faster_whisper_inference.py

@@ -7,6 +7,7 @@ from typing import BinaryIO, Union, Tuple
 from datetime import datetime, timedelta
 
 import faster_whisper
+import ctranslate2
 import whisper
 import torch
 import gradio as gr
@@ -25,7 +26,7 @@ class FasterWhisperInference(BaseInterface):
         self.available_langs = sorted(list(whisper.tokenizer.LANGUAGES.values()))
         self.translatable_models = ["large", "large-v1", "large-v2"]
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
-        self.available_compute_types = ["int8", "int8_float32", "int8_float16", "int8_bfloat16", "int16", "float16", "bfloat16", "float32"]
+        self.available_compute_types = ctranslate2.get_supported_compute_types("cuda") if self.device == "cuda" else ctranslate2.get_supported_compute_types("cpu")
         self.current_compute_type = "float16" if self.device == "cuda" else "float32"
         self.default_beam_size = 1