fix aishell3-vc0

paddlespeech/t2s/modules/tacotron2/attentions.py

@@ -171,7 +171,9 @@ class AttLoc(nn.Layer):
         if paddle.sum(att_prev) == 0:
             # if no bias, 0 0-pad goes 0
             att_prev = 1.0 - make_pad_mask(enc_hs_len)
-            att_prev = att_prev / enc_hs_len.unsqueeze(-1)
+            unsqueeze_enc_hs_len = enc_hs_len.unsqueeze(-1).astype(
+                att_prev.dtype)
+            att_prev = att_prev / unsqueeze_enc_hs_len
 
         # att_prev: (utt, frame) -> (utt, 1, 1, frame)
         # -> (utt, att_conv_chans, 1, frame)

paddlespeech/t2s/modules/tacotron2/encoder.py

@@ -162,6 +162,8 @@ class Encoder(nn.Layer):
             return xs.transpose([0, 2, 1])
         if not isinstance(ilens, paddle.Tensor):
             ilens = paddle.to_tensor(ilens)
+        if ilens.ndim == 0:
+            ilens = ilens.unsqueeze(0)
         xs = xs.transpose([0, 2, 1])
         # for dygraph to static graph
         # self.blstm.flatten_parameters()