diff options
Diffstat (limited to 'candle-examples/examples/whisper/main.rs')
| -rw-r--r-- | candle-examples/examples/whisper/main.rs | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/candle-examples/examples/whisper/main.rs b/candle-examples/examples/whisper/main.rs index f2560815..d2caebcd 100644 --- a/candle-examples/examples/whisper/main.rs +++ b/candle-examples/examples/whisper/main.rs @@ -525,7 +525,11 @@ fn main() -> Result<()> { let config: Config = serde_json::from_str(&std::fs::read_to_string(config_filename)?)?; let tokenizer = Tokenizer::from_file(tokenizer_filename).map_err(E::msg)?; - let mel_bytes = include_bytes!("melfilters.bytes"); + let mel_bytes = match config.num_mel_bins { + 80 => include_bytes!("melfilters.bytes").as_slice(), + 128 => include_bytes!("melfilters128.bytes").as_slice(), + nmel => anyhow::bail!("unexpected num_mel_bins {nmel}"), + }; let mut mel_filters = vec![0f32; mel_bytes.len() / 4]; <byteorder::LittleEndian as byteorder::ByteOrder>::read_f32_into(mel_bytes, &mut mel_filters); |