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 bug

Browse Source
This commit is contained in:
维石 2024-04-10 11:37:27 +08:00
parent 112c8e6eb7
commit b8bf792ce7
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -347,8 +347,10 @@ class CTTransformer(torch.nn.Module):
punc_array = punctuations punc_array = punctuations
else: else:
punc_array = torch.cat([punc_array, punctuations], dim=0) punc_array = torch.cat([punc_array, punctuations], dim=0)
# post processing when using word level punc model # post processing when using word level punc model
if self.jieba_usr_dict is not None: if self.jieba_usr_dict is not None:
punc_array = punc_array.reshape(-1)
len_tokens = len(tokens) len_tokens = len(tokens)
new_punc_array = copy.copy(punc_array).tolist() new_punc_array = copy.copy(punc_array).tolist()
# for i, (token, punc_id) in enumerate(zip(tokens[::-1], punc_array.tolist()[::-1])): # for i, (token, punc_id) in enumerate(zip(tokens[::-1], punc_array.tolist()[::-1])):