Fix masking bug for TTS Aligner (NVIDIA#6677)

Signed-off-by: Jocelyn Huang <[email protected]> Signed-off-by: hsiehjackson <[email protected]>
hsiehjackson · Jun 2, 2023 · d2938b9 · d2938b9
1 parent 1b93141
commit d2938b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/tts/models/aligner.py b/nemo/collections/tts/models/aligner.py
@@ -126,7 +126,7 @@ def forward(self, *, spec, spec_len, text, text_len, attn_prior=None):
             attn_soft, attn_logprob = self.alignment_encoder(
                 queries=spec,
                 keys=self.embed(text).transpose(1, 2),
-                mask=get_mask_from_lengths(text_len).unsqueeze(-1),
+                mask=get_mask_from_lengths(text_len).unsqueeze(-1) == 0,
                 attn_prior=attn_prior,
             )