bugfix: audio_len should be 1D, no 0D, which will raise list index out (

PaddlePaddle#3490) of range error in the following decode process Co-authored-by: Luzhenhui <[email protected]>
luotao1 · Jun 11, 2024 · 30b2d98 · 30b2d98
1 parent cb6fb7f
commit 30b2d98
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlespeech/cli/asr/infer.py b/paddlespeech/cli/asr/infer.py
@@ -274,7 +274,7 @@ def preprocess(self, model_type: str, input: Union[str, os.PathLike]):
             # fbank
             audio = preprocessing(audio, **preprocess_args)
 
-            audio_len = paddle.to_tensor(audio.shape[0]).unsqueeze(axis=0)
+            audio_len = paddle.to_tensor([audio.shape[0]]).unsqueeze(axis=0)
             audio = paddle.to_tensor(audio, dtype='float32').unsqueeze(axis=0)
 
             self._inputs["audio"] = audio