From d51b5523f4be794469f1c8072d8515d6b344ec7b Mon Sep 17 00:00:00 2001 From: zxcd <228587199@qq.com> Date: Wed, 15 Feb 2023 09:32:28 +0000 Subject: [PATCH] add condition ctc_prefix_beam_search decode. --- paddlespeech/s2t/models/wav2vec2/wav2vec2_ASR.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddlespeech/s2t/models/wav2vec2/wav2vec2_ASR.py b/paddlespeech/s2t/models/wav2vec2/wav2vec2_ASR.py index 059c0d909..baa7392eb 100755 --- a/paddlespeech/s2t/models/wav2vec2/wav2vec2_ASR.py +++ b/paddlespeech/s2t/models/wav2vec2/wav2vec2_ASR.py @@ -128,7 +128,7 @@ class Wav2vec2ASR(nn.Layer): # with other batch decoding mode elif decoding_method == 'ctc_prefix_beam_search': assert feats.shape[0] == 1 - if tokenizer is None: + if tokenizer is None and sb_pipeline is False: hyp = self.ctc_prefix_beam_search(feats, beam_size) res = [text_feature.defeaturize(hyp)] res_tokenids = [hyp]