fix

10 months ago · 6c2652eb7e
parent ed02bd551c
commit 6c2652eb7e
1 changed files with 4 additions and 1 deletions
--- a/paddlespeech/t2s/modules/diffnet.py
+++ b/paddlespeech/t2s/modules/diffnet.py
@ -120,7 +120,10 @@ class SinusoidalPosEmb(nn.Layer):
        self.dim = dim

    def forward(self, x: paddle.Tensor):
-        x = paddle.cast(x.unsqueeze(0), 'float32')
+        if x.ndim == 0:
+            x = paddle.cast(x.unsqueeze(0), 'float32')
+        else:
+            x = paddle.cast(x, 'float32')
        half_dim = self.dim // 2
        emb = math.log(10000) / (half_dim - 1)
        emb = paddle.exp(paddle.arange(half_dim) * -emb)