Merge pull request #2085 from yt605155624/fix_tts_cli_log

[TTS]fix int32 warning in tts
3 years ago · d8a0ba5913
parent 1ce7bfef99 c0f126ecd9
commit d8a0ba5913
1 changed files with 1 additions and 1 deletions
--- a/paddlespeech/t2s/modules/nets_utils.py
+++ b/paddlespeech/t2s/modules/nets_utils.py
@ -147,7 +147,7 @@ def make_pad_mask(lengths, xs=None, length_dim=-1):
    seq_range = paddle.arange(0, maxlen, dtype=paddle.int64)
    seq_range_expand = seq_range.unsqueeze(0).expand([bs, maxlen])
    seq_length_expand = lengths.unsqueeze(-1)
-    mask = seq_range_expand >= seq_length_expand
+    mask = seq_range_expand >= seq_length_expand.cast(seq_range_expand.dtype)

    if xs is not None:
        assert paddle.shape(xs)[0] == bs, (paddle.shape(xs)[0], bs)