Return durations at VITS inference

2026-09-10 11:50:20 +08:00 · 2022-05-11 11:30:05 +02:00
parent 5021a03de0
commit c18bd21b3f
1 changed files with 1 additions and 0 deletions
@@ -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,