未验证 提交 d8a0ba59 编写于 作者: 小湉湉's avatar 小湉湉 提交者: GitHub

Merge pull request #2085 from yt605155624/fix_tts_cli_log

[TTS]fix int32 warning in tts
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册