Add whisper large-v3 turbo to the example. (#2531)

huggingface · Oct 2, 2024 · 9363006 · 9363006
1 parent f479840
commit 9363006
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/candle-examples/examples/whisper/main.rs b/candle-examples/examples/whisper/main.rs
@@ -370,6 +370,7 @@ enum WhichModel {
  Large,
  LargeV2,
  LargeV3,
+ LargeV3Turbo,
  #[value(name = "distil-medium.en")]
  DistilMediumEn,
  #[value(name = "distil-large-v2")]
@@ -388,6 +389,7 @@ impl WhichModel {
  | Self::Large
  | Self::LargeV2
  | Self::LargeV3
+ | Self::LargeV3Turbo
  | Self::DistilLargeV2
  | Self::DistilLargeV3 => true,
  Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn | Self::DistilMediumEn => {
@@ -409,6 +411,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"),
  Self::DistilLargeV3 => ("distil-whisper/distil-large-v3", "main"),