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

Normalize loudness of the generated audio (#1795)

Browse Source 
* Normalize loudness of the generated audio.

* Lints.

* One more lint.

* Avoid running the bs1770 tests.

* Another attempt at discarding doc comments.

* Also normalize the loudness in the encodec example.
This commit is contained in:
Laurent Mazare
2024-03-03 14:00:42 +01:00
committed by GitHub
parent ca5d727ba2
commit 20abb72fec
5 changed files with 543 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-examples/examples/encodec/main.rs
View File

@ -132,7 +132,9 @@ fn main() -> Result<()> {
Action::AudioToAudio | Action::CodeToAudio => {
let pcm = model.decode(&codes)?;
println!("output pcm shape: {:?}", pcm.shape());
let pcm = pcm.i(0)?.i(0)?.to_vec1::<f32>()?;
let pcm = pcm.i(0)?.i(0)?;
let pcm = candle_examples::audio::normalize_loudness(&pcm, 24_000, true)?;
let pcm = pcm.to_vec1::<f32>()?;
let mut output = std::fs::File::create(&args.out_file)?;
candle_examples::wav::write_pcm_as_wav(&mut output, &pcm, 24_000)?;
}