huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 19:18:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support whisper large-v3 turbo in the whisper-microphone example. (#2533)

Browse Source
This commit is contained in:
Laurent Mazare
2024-10-02 22:09:14 +02:00
committed by GitHub
parent 7b60bda4ed
commit 90d04ff622
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
candle-examples/examples/whisper-microphone/main.rs
View File

@ -389,6 +389,7 @@ enum WhichModel {
Large, Large,
LargeV2, LargeV2,
LargeV3, LargeV3,
LargeV3Turbo,
#[value(name = "distil-medium.en")] #[value(name = "distil-medium.en")]
DistilMediumEn, DistilMediumEn,
#[value(name = "distil-large-v2")] #[value(name = "distil-large-v2")]
@ -405,6 +406,7 @@ impl WhichModel {
| Self::Large | Self::Large
| Self::LargeV2 | Self::LargeV2
| Self::LargeV3 | Self::LargeV3
| Self::LargeV3Turbo
| Self::DistilLargeV2 => true, | Self::DistilLargeV2 => true,
Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn | Self::DistilMediumEn => { Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn | Self::DistilMediumEn => {
false false
@ -425,6 +427,7 @@ impl WhichModel {
Self::Large => ("openai/whisper-large", "refs/pr/36"), Self::Large => ("openai/whisper-large", "refs/pr/36"),
Self::LargeV2 => ("openai/whisper-large-v2", "refs/pr/57"), Self::LargeV2 => ("openai/whisper-large-v2", "refs/pr/57"),
Self::LargeV3 => ("openai/whisper-large-v3", "main"), Self::LargeV3 => ("openai/whisper-large-v3", "main"),
Self::LargeV3Turbo => ("openai/whisper-large-v3-turbo", "main"),
Self::DistilMediumEn => ("distil-whisper/distil-medium.en", "main"), Self::DistilMediumEn => ("distil-whisper/distil-medium.en", "main"),
Self::DistilLargeV2 => ("distil-whisper/distil-large-v2", "main"), Self::DistilLargeV2 => ("distil-whisper/distil-large-v2", "main"),
} }