Skip to content

Commit e154b8f

Browse files
authored
fix (#4071)
1 parent e59a3c1 commit e154b8f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddlespeech/t2s/modules/tacotron2/attentions.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -173,6 +173,8 @@ def forward(
173173
att_prev = 1.0 - make_pad_mask(enc_hs_len)
174174
att_prev = att_prev / enc_hs_len.unsqueeze(-1).astype(
175175
att_prev.dtype)
176+
else:
177+
att_prev.unsqueeze(-1)
176178

177179
# att_prev: (utt, frame) -> (utt, 1, 1, frame)
178180
# -> (utt, att_conv_chans, 1, frame)

0 commit comments

Comments
 (0)