From 823ef51152db200303f72fc8465e53426af1bfb8 Mon Sep 17 00:00:00 2001 From: boomkin Date: Fri, 29 May 2020 13:42:44 +0200 Subject: [PATCH] byte mask fix --- flowtron.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flowtron.py b/flowtron.py index 06dea68..eac0a30 100644 --- a/flowtron.py +++ b/flowtron.py @@ -589,7 +589,7 @@ def forward(self, mel, speaker_vecs, text, in_lens, out_lens): [text, speaker_vecs.expand(text.size(0), -1, -1)], 2) log_s_list = [] attns_list = [] - mask = ~get_mask_from_lengths(in_lens)[..., None] + mask = ~get_mask_from_lengths(in_lens)[..., None].bool() for i, flow in enumerate(self.flows): mel, log_s, gate, attn = flow( mel, encoder_outputs, mask, out_lens)