diff --git a/candle-examples/examples/whisper/main.rs b/candle-examples/examples/whisper/main.rs index 07247451..4007bd8d 100644 --- a/candle-examples/examples/whisper/main.rs +++ b/candle-examples/examples/whisper/main.rs @@ -361,15 +361,25 @@ enum WhichModel { MediumEn, Large, LargeV2, + #[value(name = "distil-medium.en")] + DistilMediumEn, + #[value(name = "distil-large-v2")] + DistilLargeV2, } impl WhichModel { fn is_multilingual(&self) -> bool { match self { - Self::Tiny | Self::Base | Self::Small | Self::Medium | Self::Large | Self::LargeV2 => { - true + Self::Tiny + | Self::Base + | Self::Small + | Self::Medium + | Self::Large + | Self::LargeV2 + | Self::DistilLargeV2 => true, + Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn | Self::DistilMediumEn => { + false } - Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn => false, } } @@ -385,6 +395,8 @@ impl WhichModel { Self::MediumEn => ("openai/whisper-medium.en", "main"), Self::Large => ("openai/whisper-large", "refs/pr/36"), Self::LargeV2 => ("openai/whisper-large-v2", "refs/pr/57"), + Self::DistilMediumEn => ("distil-whisper/distil-medium.en", "main"), + Self::DistilLargeV2 => ("distil-whisper/distil-large-v2", "main"), } } }