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

Bugfix for the mel filters.

Browse Source
This commit is contained in:
laurent
2023-07-05 12:56:04 +01:00
parent 4df250cf37
commit d560855c2a
2 changed files with 13 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
candle-examples/examples/whisper/audio.rs
View File

@ -148,7 +148,7 @@ fn log_mel_spectrogram_w<T: Float>(
mel
}
fn log_mel_spectrogram_<T: Float>(
fn log_mel_spectrogram_<T: Float + std::fmt::Display>(
samples: &[T],
filters: &[T],
fft_size: usize,
@ -200,20 +200,17 @@ fn log_mel_spectrogram_<T: Float>(
mel
}
pub fn pcm_to_mel<T: Float>(
pub fn pcm_to_mel<T: Float + std::fmt::Display>(
samples: &[T],
filters: &[T],
n_mel: usize,
n_fft: usize,
) -> anyhow::Result<Vec<T>> {
if filters.len() != n_mel * n_fft {
anyhow::bail!(
"unexpected filter length {} (n_mel: {}, n_fft: {})",
filters.len(),
n_mel,
n_fft
)
}
let mel = log_mel_spectrogram_(samples, filters, n_fft, super::HOP_LENGTH, n_mel, false);
let mel = log_mel_spectrogram_(
samples,
filters,
super::N_FFT,
super::HOP_LENGTH,
super::N_MELS,
false,
);
Ok(mel)
}