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

Load the mel filters.

Browse Source
This commit is contained in:
laurent
2023-07-05 11:20:33 +01:00
parent c701ee33a7
commit 26d1a7803f
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -59,6 +59,12 @@ struct Args {
/// The seed to use when generating random samples.
#[arg(long, default_value_t = 299792458)]
seed: u64,
#[arg(
long,
default_value = "candle-examples/examples/whisper/mel_filters.safetensors"
)]
filters: String,
}
#[derive(Debug, Clone)]
@ -189,6 +195,11 @@ fn main() -> Result<()> {
let tokenizer = Tokenizer::from_file(args.tokenizer_config).map_err(E::msg)?;
let filters = unsafe { candle::safetensors::MmapedFile::new(args.filters)? };
let filters = filters.deserialize()?;
let filters = filters.tensor("mel_80", &device)?;
println!("loaded mel filters {:?}", filters.shape());
let input = unsafe { candle::safetensors::MmapedFile::new(args.input)? };
let input = input.deserialize()?;
let mel = input.tensor("mel", &device)?;