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

Browse Source
This commit is contained in:
haoneng.lhn 2023-03-29 13:28:00 +08:00
parent d0d8684b96
commit a65a408f28
1 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
funasr/models/predictor/cif.py
View File

@ -228,13 +228,6 @@ class CifPredictorV2(nn.Module):
if cache["cif_alphas"] is not None:
alphas = torch.cat((cache["cif_alphas"], alphas), -1)
#if cache["is_final"]:
# tail_threshold = torch.tensor([self.tail_threshold], dtype=alphas.dtype).to(alphas.device)
# tail_threshold = torch.reshape(tail_threshold, (1, 1))
# alphas = torch.cat([alphas, tail_threshold], dim=1)
# zeros_hidden = torch.zeros((b, 1, d), dtype=hidden.dtype).to(hidden.device)
# hidden = torch.cat([hidden, zeros_hidden], dim=1)
token_num = alphas.sum(-1)
acoustic_embeds, cif_peak = cif(hidden, alphas, self.threshold)
len_time = alphas.size(-1)