speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix decoding_ind none bug

Browse Source
This commit is contained in:
haoneng.lhn 2023-09-12 13:00:16 +08:00
parent eed5cbb945
commit 2165d5de05
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/bin/asr_infer.py
View File

@ -430,9 +430,7 @@ class Speech2TextParaformer:
# b. Forward Encoder
if decoding_ind is None:
decoding_ind = self.decoding_ind
if self.decoding_ind is None:
decoding_ind = 0
decoding_ind = 0 if self.decoding_ind is None else self.decoding_ind
enc, enc_len = self.asr_model.encode(**batch, ind=decoding_ind)
if isinstance(enc, tuple):
enc = enc[0]