From 06c2993dfceadf05a48e9fd469ac455fe5302624 Mon Sep 17 00:00:00 2001 From: zr_jin Date: Tue, 5 Nov 2024 11:45:32 +0800 Subject: [PATCH] Update infer.py --- egs/ljspeech/TTS/matcha/infer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/ljspeech/TTS/matcha/infer.py b/egs/ljspeech/TTS/matcha/infer.py index 06671b7ec..bcf9a8b4a 100755 --- a/egs/ljspeech/TTS/matcha/infer.py +++ b/egs/ljspeech/TTS/matcha/infer.py @@ -183,13 +183,13 @@ def infer_dataset( sf.write( file=params.save_wav_dir / f"{cut_ids[i]}_pred.wav", data=output["waveform"], - samplerate=params.sampling_rate, + samplerate=params.data_args.sampling_rate, subtype="PCM_16", ) sf.write( file=params.save_wav_dir / f"{cut_ids[i]}_gt.wav", data=audio[i].numpy(), - samplerate=params.sampling_rate, + samplerate=params.data_args.sampling_rate, subtype="PCM_16", )