From bc5db13d067332baecb932f54fe4abe5398be016 Mon Sep 17 00:00:00 2001 From: Edresson Casanova Date: Sat, 19 Feb 2022 19:24:00 +0000 Subject: [PATCH] Fix the bug in extract tts spectrogram script --- TTS/bin/extract_tts_spectrograms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TTS/bin/extract_tts_spectrograms.py b/TTS/bin/extract_tts_spectrograms.py index 7b489fd6..386cf332 100755 --- a/TTS/bin/extract_tts_spectrograms.py +++ b/TTS/bin/extract_tts_spectrograms.py @@ -138,7 +138,7 @@ def inference( aux_input={"d_vectors": speaker_c, "speaker_ids": speaker_ids}, ) model_output = outputs["model_outputs"] - model_output = model_output.transpose(1, 2).detach().cpu().numpy() + model_output = model_output.detach().cpu().numpy() elif "tacotron" in model_name: aux_input = {"speaker_ids": speaker_ids, "d_vectors": d_vectors}