From 26fd37b34854bd2b6bab97309d3ea9585921189f Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Fri, 18 Aug 2023 08:18:30 +0100 Subject: [PATCH] Use the main branch of the HF repo where possible. (#498) * Use the main branch of the HF repo where possible. * And add the large model. --- candle-examples/examples/whisper/main.rs | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/candle-examples/examples/whisper/main.rs b/candle-examples/examples/whisper/main.rs index 74b02e64..4788385b 100644 --- a/candle-examples/examples/whisper/main.rs +++ b/candle-examples/examples/whisper/main.rs @@ -283,13 +283,16 @@ enum WhichModel { Medium, #[value(name = "medium.en")] MediumEn, + Large, LargeV2, } impl WhichModel { fn is_multilingual(&self) -> bool { match self { - Self::Tiny | Self::Base | Self::Small | Self::Medium | Self::LargeV2 => true, + Self::Tiny | Self::Base | Self::Small | Self::Medium | Self::Large | Self::LargeV2 => { + true + } Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn => false, } } @@ -301,8 +304,9 @@ impl WhichModel { Self::BaseEn => ("openai/whisper-base.en", "refs/pr/13"), Self::Small => ("openai/whisper-small", "main"), Self::SmallEn => ("openai/whisper-small.en", "refs/pr/10"), - Self::Medium => ("openai/whisper-medium", "refs/pr/21"), - Self::MediumEn => ("openai/whisper-medium.en", "refs/pr/11"), + Self::Medium => ("openai/whisper-medium", "main"), + Self::MediumEn => ("openai/whisper-medium.en", "main"), + Self::Large => ("openai/whisper-large", "refs/pr/36"), Self::LargeV2 => ("openai/whisper-large-v2", "refs/pr/57"), } }