diff options
| author | Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com> | 2024-03-21 16:16:49 +0530 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-21 11:46:49 +0100 |
| commit | 0c11e055be1e156153b71cff8ea714d5829b9371 (patch) | |
| tree | 2fa53e291480a2d334a4cdb04e88cb731d3c703b /candle-examples | |
| parent | 18036c6ccbd216e0397fa6469bb50a8b8ea47892 (diff) | |
| download | candle-0c11e055be1e156153b71cff8ea714d5829b9371.tar.gz candle-0c11e055be1e156153b71cff8ea714d5829b9371.tar.bz2 candle-0c11e055be1e156153b71cff8ea714d5829b9371.zip | |
support distil-large-v3 (#1898)
Diffstat (limited to 'candle-examples')
| -rw-r--r-- | candle-examples/examples/whisper/main.rs | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/candle-examples/examples/whisper/main.rs b/candle-examples/examples/whisper/main.rs index da8c73ae..ecd5ff84 100644 --- a/candle-examples/examples/whisper/main.rs +++ b/candle-examples/examples/whisper/main.rs @@ -374,6 +374,8 @@ enum WhichModel { DistilMediumEn, #[value(name = "distil-large-v2")] DistilLargeV2, + #[value(name = "distil-large-v3")] + DistilLargeV3, } impl WhichModel { @@ -386,7 +388,8 @@ impl WhichModel { | Self::Large | Self::LargeV2 | Self::LargeV3 - | Self::DistilLargeV2 => true, + | Self::DistilLargeV2 + | Self::DistilLargeV3 => true, Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn | Self::DistilMediumEn => { false } @@ -408,6 +411,7 @@ impl WhichModel { Self::LargeV3 => ("openai/whisper-large-v3", "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"), } } } |