Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix return gpt_latents

This commit is contained in:
WeberJulian 2023-11-15 14:37:53 +01:00
parent d64cafccd2
commit 83988c922a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
TTS/tts/models/xtts.py
View File

@ -568,11 +568,12 @@ class Xtts(BaseTTS):
mode="linear" mode="linear"
).transpose(1, 2) ).transpose(1, 2)
gpt_latents_list.append(gpt_latents.cpu())
wavs.append(self.hifigan_decoder(gpt_latents, g=speaker_embedding).cpu().squeeze()) wavs.append(self.hifigan_decoder(gpt_latents, g=speaker_embedding).cpu().squeeze())
return { return {
"wav": torch.cat(wavs, dim=0).numpy(), "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, "speaker_embedding": speaker_embedding,
} }