fix input dtype of elementwise_mul op from bool to int64 (#3054)

3 years ago · 0e820fc686
parent 2844112ab2
commit 0e820fc686
1 changed files with 3 additions and 2 deletions
--- a/paddlespeech/t2s/models/vits/generator.py
+++ b/paddlespeech/t2s/models/vits/generator.py
@ -559,8 +559,9 @@ class VITSGenerator(nn.Layer):
            y_lengths = paddle.cast(
                paddle.clip(paddle.sum(dur, [1, 2]), min=1), dtype='int64')
            y_mask = make_non_pad_mask(y_lengths).unsqueeze(1)
-            attn_mask = paddle.unsqueeze(x_mask, 2) * paddle.unsqueeze(y_mask,
-                                                                       -1)
+            tmp_a = paddle.cast(paddle.unsqueeze(x_mask, 2), dtype='int64')
+            tmp_b = paddle.cast(paddle.unsqueeze(y_mask, -1), dtype='int64')
+            attn_mask = tmp_a * tmp_b
            attn = self._generate_path(dur, attn_mask)

            # expand the length to match with the feature sequence