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

Merge pull request #527 from alibaba-damo-academy/dev_lhn

Browse Source 
fix paraformer online fbank cache bug
This commit is contained in:
hnluo 2023-05-18 19:53:13 +08:00 committed by GitHub
commit 193d47d829
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
funasr/bin/asr_infer.py
View File

@ -749,10 +749,13 @@ class Speech2TextParaformerOnline:
feats = cache_en["feats"]
feats_len = torch.tensor([feats.shape[1]])
self.asr_model.frontend = None
self.frontend.cache_reset()
results = self.infer(feats, feats_len, cache)
return results
else:
if self.frontend is not None:
if cache_en["start_idx"] == 0:
self.frontend.cache_reset()
feats, feats_len = self.frontend.forward(speech, speech_lengths, cache_en["is_final"])
feats = to_device(feats, device=self.device)
feats_len = feats_len.int()