From a4f061eb1c1ae5e51081e765cad101f5d141600a Mon Sep 17 00:00:00 2001 From: Jordi Mas Date: Thu, 10 Oct 2024 11:41:37 +0200 Subject: [PATCH] Add support for large-v3-turbo model --- src/whisper_ctranslate2/commandline.py | 1 + src/whisper_ctranslate2/whisper_ctranslate2.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/src/whisper_ctranslate2/commandline.py b/src/whisper_ctranslate2/commandline.py index 2bd69c8..bf4e6aa 100644 --- a/src/whisper_ctranslate2/commandline.py +++ b/src/whisper_ctranslate2/commandline.py @@ -14,6 +14,7 @@ "large-v1", "large-v2", "large-v3", + "large-v3-turbo", "distil-large-v2", "distil-large-v3", "distil-medium.en", diff --git a/src/whisper_ctranslate2/whisper_ctranslate2.py b/src/whisper_ctranslate2/whisper_ctranslate2.py index 47e7c5f..2194fdb 100644 --- a/src/whisper_ctranslate2/whisper_ctranslate2.py +++ b/src/whisper_ctranslate2/whisper_ctranslate2.py @@ -112,6 +112,9 @@ def main(): hf_token = args.pop("hf_token") speaker_name = args.pop("speaker_name") + if model == "large-v3-turbo": + model = "deepdml/faster-whisper-large-v3-turbo-ct2" + language = get_language(language, model_directory, model) options = get_transcription_options(args)