Fix return gpt_latents

coqui-ai · Nov 15, 2023 · 83988c9 · 83988c9
1 parent d64cafc
commit 83988c9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/TTS/tts/models/xtts.py b/TTS/tts/models/xtts.py
@@ -568,11 +568,12 @@ def inference(
                         mode="linear"
                     ).transpose(1, 2)
 
+                gpt_latents_list.append(gpt_latents.cpu())
                 wavs.append(self.hifigan_decoder(gpt_latents, g=speaker_embedding).cpu().squeeze())
 
         return {
             "wav": torch.cat(wavs, dim=0).numpy(),
-            "gpt_latents": torch.cat(gpt_latents_list, dim=0).numpy(),
+            "gpt_latents": torch.cat(gpt_latents_list, dim=1).numpy(),
             "speaker_embedding": speaker_embedding,
         }