Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -45,7 +45,8 @@ class WhisperBase(ABC):
         self.current_model_size = None
         self.available_models = whisper.available_models()
         self.available_langs = sorted(list(whisper.tokenizer.LANGUAGES.values()))
-        self.translatable_models = ["large", "large-v1", "large-v2", "large-v3"]
+        #self.translatable_models = ["large", "large-v1", "large-v2", "large-v3"]
+        self.translatable_models = whisper.available_models()
         self.device = self.get_device()
         self.available_compute_types = ["float16", "float32"]
         self.current_compute_type = "float16" if self.device == "cuda" else "float32"