From 6c2652eb7eca7a7059f397a1c045f9c3d3e23c43 Mon Sep 17 00:00:00 2001 From: enkilee Date: Thu, 5 Dec 2024 09:27:05 +0800 Subject: [PATCH] fix --- paddlespeech/t2s/modules/diffnet.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/paddlespeech/t2s/modules/diffnet.py b/paddlespeech/t2s/modules/diffnet.py index 30a49ce7a..607fbd19d 100644 --- 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)