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

Update sanm_encoder.py

Browse Source
This commit is contained in:
hnluo 2023-04-27 01:49:01 +08:00 committed by GitHub
parent c5992ca03e
commit 9ff5b683db
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
funasr/models/encoder/sanm_encoder.py
View File

@ -379,7 +379,10 @@ class SANMEncoder(AbsEncoder):
xs_pad = xs_pad xs_pad = xs_pad
else: else:
xs_pad = self.embed(xs_pad, cache) xs_pad = self.embed(xs_pad, cache)
xs_pad = self._add_overlap_chunk(xs_pad, cache) if cache["tail_chunk"]:
xs_pad = cache["feats"]
else:
xs_pad = self._add_overlap_chunk(xs_pad, cache)
encoder_outs = self.encoders0(xs_pad, None, None, None, None) encoder_outs = self.encoders0(xs_pad, None, None, None, None)
xs_pad, masks = encoder_outs[0], encoder_outs[1] xs_pad, masks = encoder_outs[0], encoder_outs[1]
intermediate_outs = [] intermediate_outs = []