From f056dcab2153ae3719a87c741e4591cf46bcae67 Mon Sep 17 00:00:00 2001
From: Franco Lucchini <flucchini@gmail.com>
Date: Fri, 18 Aug 2023 09:08:59 +0200
Subject: [PATCH] Add medium model (#497)

---
 candle-examples/examples/whisper/main.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/candle-examples/examples/whisper/main.rs b/candle-examples/examples/whisper/main.rs
index 4ea60fb4..74b02e64 100644
--- a/candle-examples/examples/whisper/main.rs
+++ b/candle-examples/examples/whisper/main.rs
@@ -280,6 +280,7 @@ enum WhichModel {
     Small,
     #[value(name = "small.en")]
     SmallEn,
+    Medium,
     #[value(name = "medium.en")]
     MediumEn,
     LargeV2,
@@ -288,7 +289,7 @@ enum WhichModel {
 impl WhichModel {
     fn is_multilingual(&self) -> bool {
         match self {
-            Self::Tiny | Self::Base | Self::Small | Self::LargeV2 => true,
+            Self::Tiny | Self::Base | Self::Small | Self::Medium | Self::LargeV2 => true,
             Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn => false,
         }
     }
@@ -300,6 +301,7 @@ 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::LargeV2 => ("openai/whisper-large-v2", "refs/pr/57"),
         }