Update infer.py

k2-fsa · Nov 5, 2024 · 06c2993 · 06c2993
1 parent f9056be
commit 06c2993
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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",
             )