fix slice bug of speedyspeech expand, test=tts (#1337)

3 years ago · a1867c20c3
parent 2170a5cb1d
commit a1867c20c3
1 changed files with 3 additions and 1 deletions
--- a/paddlespeech/t2s/models/speedyspeech/speedyspeech.py
+++ b/paddlespeech/t2s/models/speedyspeech/speedyspeech.py
@ -30,7 +30,9 @@ def expand(encodings: paddle.Tensor, durations: paddle.Tensor) -> paddle.Tensor:
        k = 0
        for j in range(t_enc):
            d = durations[i, j]
-            M[i, k:k + d, j] = 1
+            # If the d == 0, slice action is meaningless and not supported
+            if d >= 1:
+                M[0, k:k + d, j] = 1
            k += d
    encodings = paddle.matmul(M, encodings)
    return encodings