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

Use the main branch of the HF repo where possible. (#498)

Browse Source 
* Use the main branch of the HF repo where possible.

* And add the large model.
This commit is contained in:
Laurent Mazare
2023-08-18 08:18:30 +01:00
committed by GitHub
parent f056dcab21
commit 26fd37b348
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
candle-examples/examples/whisper/main.rs
View File

@ -283,13 +283,16 @@ enum WhichModel {
Medium, Medium,
#[value(name = "medium.en")] #[value(name = "medium.en")]
MediumEn, MediumEn,
Large,
LargeV2, LargeV2,
} }
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::LargeV2 => true, Self::Tiny | Self::Base | Self::Small | Self::Medium | Self::Large | Self::LargeV2 => {
true
}
Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn => false, Self::TinyEn | Self::BaseEn | Self::SmallEn | Self::MediumEn => false,
} }
} }
@ -301,8 +304,9 @@ impl WhichModel {
Self::BaseEn => ("openai/whisper-base.en", "refs/pr/13"), Self::BaseEn => ("openai/whisper-base.en", "refs/pr/13"),
Self::Small => ("openai/whisper-small", "main"), Self::Small => ("openai/whisper-small", "main"),
Self::SmallEn => ("openai/whisper-small.en", "refs/pr/10"), Self::SmallEn => ("openai/whisper-small.en", "refs/pr/10"),
Self::Medium => ("openai/whisper-medium", "refs/pr/21"), Self::Medium => ("openai/whisper-medium", "main"),
Self::MediumEn => ("openai/whisper-medium.en", "refs/pr/11"), Self::MediumEn => ("openai/whisper-medium.en", "main"),
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"),
} }
} }