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

Return durations at VITS inference

This commit is contained in:
Eren Gölge 2022-05-11 11:30:05 +02:00
parent 5021a03de0
commit c18bd21b3f
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
TTS/tts/models/vits.py
View File

@ -1054,6 +1054,7 @@ class Vits(BaseTTS):
outputs = { outputs = {
"model_outputs": o, "model_outputs": o,
"alignments": attn.squeeze(1), "alignments": attn.squeeze(1),
"durations": w_ceil,
"z": z, "z": z,
"z_p": z_p, "z_p": z_p,
"m_p": m_p, "m_p": m_p,