mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 10:38:54 +00:00
Add support for distil whisper (#1245)
* Add support for distil-whisper. * Add distil-large. * Rename the large model.
This commit is contained in:
@ -361,15 +361,25 @@ enum WhichModel {
|
|||||||
MediumEn,
|
MediumEn,
|
||||||
Large,
|
Large,
|
||||||
LargeV2,
|
LargeV2,
|
||||||
|
#[value(name = "distil-medium.en")]
|
||||||
|
DistilMediumEn,
|
||||||
|
#[value(name = "distil-large-v2")]
|
||||||
|
DistilLargeV2,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl WhichModel {
|
impl WhichModel {
|
||||||
fn is_multilingual(&self) -> bool {
|
fn is_multilingual(&self) -> bool {
|
||||||
match self {
|
match self {
|
||||||
Self::Tiny | Self::Base | Self::Small | Self::Medium | Self::Large | Self::LargeV2 => {
|
Self::Tiny
|
||||||
true
|
| Self::Base
|
||||||
|
| Self::Small
|
||||||
|
| Self::Medium
|
||||||
|
| Self::Large
|
||||||
|
| Self::LargeV2
|
||||||
|
| Self::DistilLargeV2 => true,
|
||||||
|
Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn | Self::DistilMediumEn => {
|
||||||
|
false
|
||||||
}
|
}
|
||||||
Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn => false,
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -385,6 +395,8 @@ impl WhichModel {
|
|||||||
Self::MediumEn => ("openai/whisper-medium.en", "main"),
|
Self::MediumEn => ("openai/whisper-medium.en", "main"),
|
||||||
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::DistilMediumEn => ("distil-whisper/distil-medium.en", "main"),
|
||||||
|
Self::DistilLargeV2 => ("distil-whisper/distil-large-v2", "main"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user