From bce4dc10d2f2f6b972a476c2afe22eaf83b3c8ef Mon Sep 17 00:00:00 2001 From: Laurent Date: Wed, 2 Oct 2024 21:35:30 +0200 Subject: [PATCH] Support whisper large-v3 turbo in the whisper-microphone example. --- candle-examples/examples/whisper-microphone/main.rs | 3 +++ 1 file changed, 3 insertions(+) diff --git a/candle-examples/examples/whisper-microphone/main.rs b/candle-examples/examples/whisper-microphone/main.rs index 44a64b05a8..5165da1c1e 100644 --- a/candle-examples/examples/whisper-microphone/main.rs +++ b/candle-examples/examples/whisper-microphone/main.rs @@ -389,6 +389,7 @@ enum WhichModel { Large, LargeV2, LargeV3, + LargeV3Turbo, #[value(name = "distil-medium.en")] DistilMediumEn, #[value(name = "distil-large-v2")] @@ -405,6 +406,7 @@ impl WhichModel { | Self::Large | Self::LargeV2 | Self::LargeV3 + | Self::LargeV3Turbo | Self::DistilLargeV2 => true, Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn | Self::DistilMediumEn => { false @@ -425,6 +427,7 @@ impl WhichModel { Self::Large => ("openai/whisper-large", "refs/pr/36"), Self::LargeV2 => ("openai/whisper-large-v2", "refs/pr/57"), Self::LargeV3 => ("openai/whisper-large-v3", "main"), + Self::LargeV3Turbo => ("openai/whisper-large-v3-turbo", "main"), Self::DistilMediumEn => ("distil-whisper/distil-medium.en", "main"), Self::DistilLargeV2 => ("distil-whisper/distil-large-v2", "main"), }