rearrange-encoder-infer-param (#2983)

3 years ago · c02bc087f6
parent 367d8293fa
commit c02bc087f6
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