From 52efb7eb3d91f25e78c16dc006a8f94143b5206f Mon Sep 17 00:00:00 2001 From: Laurent <laurent.mazare@gmail.com> Date: Thu, 2 Nov 2023 19:28:30 +0100 Subject: [PATCH] Rename the large model. --- candle-examples/examples/whisper/main.rs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/candle-examples/examples/whisper/main.rs b/candle-examples/examples/whisper/main.rs index 1aa7ec3b56..4007bd8d56 100644 --- a/candle-examples/examples/whisper/main.rs +++ b/candle-examples/examples/whisper/main.rs @@ -363,8 +363,8 @@ enum WhichModel { LargeV2, #[value(name = "distil-medium.en")] DistilMediumEn, - #[value(name = "distil-large")] - DistilLarge, + #[value(name = "distil-large-v2")] + DistilLargeV2, } impl WhichModel { @@ -376,7 +376,7 @@ impl WhichModel { | Self::Medium | Self::Large | Self::LargeV2 - | Self::DistilLarge => true, + | Self::DistilLargeV2 => true, Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn | Self::DistilMediumEn => { false } @@ -396,7 +396,7 @@ impl WhichModel { 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::DistilLarge => ("distil-whisper/distil-large-v2", "main"), + Self::DistilLargeV2 => ("distil-whisper/distil-large-v2", "main"), } } }