Fix the bug in extract tts spectrogram script

2022-02-19 19:24:00 +00:00 · 2022-02-19 19:24:00 +00:00 · bc5db13d06
parent 05fffb0ebc
commit bc5db13d06
1 changed files with 1 additions and 1 deletions
--- 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}