Fix return gpt_latents

coqui-ai · Nov 15, 2023 · 7cd4e24 · 7cd4e24
1 parent a8df727
commit 7cd4e24
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/TTS/tts/models/xtts.py b/TTS/tts/models/xtts.py
@@ -599,11 +599,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,
         }