rearrange-encoder-infer-param

Signed-off-by: 443127316 <443127316@qq.com>
pull/2983/head
443127316 3 years ago
parent bc64f6eefd
commit 0b671997d8

@ -705,9 +705,9 @@ class FastSpeech2(nn.Layer):
def encoder_infer(
self,
text: paddle.Tensor,
spk_id=None,
alpha: float=1.0,
spk_emb=None,
spk_id=None,
tone_id=None,
) -> Tuple[paddle.Tensor, paddle.Tensor, paddle.Tensor]:
# input of embedding must be int64

Loading…
Cancel
Save