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

bug fix

Browse Source
This commit is contained in:
shixian.shi 2024-03-06 15:43:29 +08:00
parent e451eb799a
commit 5b2218076b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/models/ct_transformer/model.py
View File

@ -359,7 +359,7 @@ class CTTransformer(torch.nn.Module):
ind_append = len_tokens - i - 1 ind_append = len_tokens - i - 1
for _ in range(num_append): for _ in range(num_append):
new_punc_array.insert(ind_append, 1) new_punc_array.insert(ind_append, 1)
punc_array = torch.tensor(new_punc_array) punc_array = torch.tensor(new_punc_array)
result_i = {"key": key[0], "text": new_mini_sentence_out, "punc_array": punc_array} result_i = {"key": key[0], "text": new_mini_sentence_out, "punc_array": punc_array}
results.append(result_i) results.append(result_i)