From ed02bd551c23b8e833d5aa77962a52a5da7c71a7 Mon Sep 17 00:00:00 2001 From: enkilee Date: Mon, 2 Dec 2024 11:53:19 +0800 Subject: [PATCH] fix --- paddlespeech/t2s/modules/nets_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddlespeech/t2s/modules/nets_utils.py b/paddlespeech/t2s/modules/nets_utils.py index 755190200..e32be0706 100644 --- a/paddlespeech/t2s/modules/nets_utils.py +++ b/paddlespeech/t2s/modules/nets_utils.py @@ -182,7 +182,7 @@ def make_pad_mask(lengths, xs=None, length_dim=-1): raise ValueError("length_dim cannot be 0: {}".format(length_dim)) if lengths.ndim == 0: - bs = paddle.shape(lengths) + bs = paddle.shape(lengths.unsqueeze(0)) else: bs = paddle.shape(lengths)