mirror of https://github.com/coqui-ai/TTS.git
pylint
This commit is contained in:
parent
bdf6944665
commit
eb51d5409a
|
@ -177,8 +177,7 @@ class TacotronAbstract(ABC, nn.Module):
|
||||||
elif style_input is None:
|
elif style_input is None:
|
||||||
gst_outputs = torch.zeros(1, 1, self.gst_embedding_dim).to(device)
|
gst_outputs = torch.zeros(1, 1, self.gst_embedding_dim).to(device)
|
||||||
else:
|
else:
|
||||||
# pylint: disable=not-callable
|
gst_outputs = self.gst_layer(style_input) # pylint: disable=not-callable
|
||||||
gst_outputs = self.gst_layer(style_input)
|
|
||||||
embedded_gst = gst_outputs.repeat(1, inputs.size(1), 1)
|
embedded_gst = gst_outputs.repeat(1, inputs.size(1), 1)
|
||||||
return inputs, embedded_gst
|
return inputs, embedded_gst
|
||||||
|
|
||||||
|
|
|
@ -60,6 +60,5 @@
|
||||||
"gst_embedding_dim": 512,
|
"gst_embedding_dim": 512,
|
||||||
"gst_num_heads": 4,
|
"gst_num_heads": 4,
|
||||||
"gst_style_tokens": 10
|
"gst_style_tokens": 10
|
||||||
},
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue