From c18bd21b3f4fdd4c9686fa124a3e289998a6445e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Wed, 11 May 2022 11:30:05 +0200
Subject: [PATCH] Return durations at VITS inference

---
 TTS/tts/models/vits.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py
index 1c623f50..d0ee214f 100644
--- 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,