fix fft_size key error

2020-07-12 10:40:33 +02:00 · 2020-07-12 10:40:33 +02:00 · 447176258c
parent c2d8a338a1
commit 447176258c
1 changed files with 1 additions and 0 deletions
--- a/mozilla_voice_tts/tts/models/tacotron_abstract.py
+++ b/mozilla_voice_tts/tts/models/tacotron_abstract.py
@ -177,6 +177,7 @@ class TacotronAbstract(ABC, nn.Module):
        elif style_input is None:
            gst_outputs = torch.zeros(1, 1, self.gst_embedding_dim).to(device)
        else:
+            # pylint: disable=not-callable
            gst_outputs = self.gst_layer(style_input)
        embedded_gst = gst_outputs.repeat(1, inputs.size(1), 1)
        return inputs, embedded_gst