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

Browse Source
This commit is contained in:
九耳 2023-03-30 16:21:44 +08:00
parent 7df8452a85
commit 3266665f3d
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
funasr/runtime/python/onnxruntime/funasr_onnx/punc_bin.py
View File

@ -101,8 +101,7 @@ class TargetDelayTransformer():
mini_sentence = mini_sentence[0:sentenceEnd + 1] mini_sentence = mini_sentence[0:sentenceEnd + 1]
punctuations = punctuations[0:sentenceEnd + 1] punctuations = punctuations[0:sentenceEnd + 1]
punctuations_np = punctuations.cpu().numpy() new_mini_sentence_punc += [int(x) for x in punctuations]
new_mini_sentence_punc += [int(x) for x in punctuations_np]
words_with_punc = [] words_with_punc = []
for i in range(len(mini_sentence)): for i in range(len(mini_sentence)):
if i > 0: if i > 0: