rearrange-encoder-infer-param

Signed-off-by: 443127316 <443127316@qq.com>
3 years ago · 0b671997d8
parent bc64f6eefd
commit 0b671997d8
1 changed files with 1 additions and 1 deletions
--- a/paddlespeech/t2s/models/fastspeech2/fastspeech2.py
+++ b/paddlespeech/t2s/models/fastspeech2/fastspeech2.py
@ -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