Return durations at VITS inference

2022-05-11 11:30:05 +02:00 · 2022-05-11 11:30:05 +02:00 · c18bd21b3f
parent 5021a03de0
commit c18bd21b3f
1 changed files with 1 additions and 0 deletions
--- a/TTS/tts/models/vits.py
+++ b/TTS/tts/models/vits.py
@ -1054,6 +1054,7 @@ class Vits(BaseTTS):
        outputs = {
            "model_outputs": o,
            "alignments": attn.squeeze(1),
+            "durations": w_ceil,
            "z": z,
            "z_p": z_p,
            "m_p": m_p,