diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-10-02 21:07:08 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-10-02 21:07:08 +0200 |
| commit | 936300678d588c6525594ef2578737e0c19ecf07 (patch) | |
| tree | 4109b36681930a07ba562d14ba7e226813a3d3f2 /candle-examples/examples | |
| parent | f479840ce6d2222bd004b6f275494297f1f0ae91 (diff) | |
| download | candle-936300678d588c6525594ef2578737e0c19ecf07.tar.gz candle-936300678d588c6525594ef2578737e0c19ecf07.tar.bz2 candle-936300678d588c6525594ef2578737e0c19ecf07.zip | |
Add whisper large-v3 turbo to the example. (#2531)
Diffstat (limited to 'candle-examples/examples')
| -rw-r--r-- | candle-examples/examples/whisper/main.rs | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/candle-examples/examples/whisper/main.rs b/candle-examples/examples/whisper/main.rs index ecd5ff84..84aa8b74 100644 --- 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"), |