From 8e5f98476e422c17a4db4d7d2b1143e9c380fc1f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jim=20O=E2=80=99Regan?= Date: Mon, 2 Dec 2024 15:21:31 +0100 Subject: [PATCH] Fix a typo --- matcha/models/matcha_tts.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/matcha/models/matcha_tts.py b/matcha/models/matcha_tts.py index 092fa27..31d2557 100644 --- a/matcha/models/matcha_tts.py +++ b/matcha/models/matcha_tts.py @@ -106,6 +106,7 @@ class MatchaTTS(BaseLightningClass): # 🍵 # Lengths of mel spectrograms "rtf": float, # Real-time factor + } """ # For RTF computation t = dt.datetime.now() @@ -152,7 +153,7 @@ class MatchaTTS(BaseLightningClass): # 🍵 def forward(self, x, x_lengths, y, y_lengths, spks=None, out_size=None, cond=None, durations=None): """ Computes 3 losses: - 1. duration loss: loss between predicted token durations and those extracted by Monotinic Alignment Search (MAS). + 1. duration loss: loss between predicted token durations and those extracted by Monotonic Alignment Search (MAS). 2. prior loss: loss between mel-spectrogram and encoder outputs. 3. flow matching loss: loss between mel-spectrogram and decoder outputs.