Adding the possibility of get durations out of pretrained model

2026-02-05 18:29:19 +08:00 · 2024-05-24 11:34:51 +02:00
parent fb7b954de5
commit 4b39f6cad0
13 changed files with 274 additions and 24 deletions
--- a/matcha/models/matcha_tts.py
+++ b/matcha/models/matcha_tts.py
@@ -236,4 +236,4 @@ class MatchaTTS(BaseLightningClass):  # 🍵
        else:
            prior_loss = 0

-        return dur_loss, prior_loss, diff_loss
+        return dur_loss, prior_loss, diff_loss, attn