Update infer.py

k2-fsa · Nov 5, 2024 · f9056be · f9056be
1 parent 8c6141e
commit f9056be
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
@@ -181,13 +181,13 @@ def infer_dataset(
             output["waveform"] = to_waveform(output["mel"], vocoder, denoiser)
 
             sf.write(
-                file=params.save_wave_dir / f"{cut_ids[i]}_pred.wav",
+                file=params.save_wav_dir / f"{cut_ids[i]}_pred.wav",
                 data=output["waveform"],
                 samplerate=params.sampling_rate,
                 subtype="PCM_16",
             )
             sf.write(
-                file=params.save_wave_dir / f"{cut_ids[i]}_gt.wav",
+                file=params.save_wav_dir / f"{cut_ids[i]}_gt.wav",
                 data=audio[i].numpy(),
                 samplerate=params.sampling_rate,
                 subtype="PCM_16",