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

sdpa bugfix

Browse Source
This commit is contained in:
游雁 2024-07-24 00:40:52 +08:00
parent b9bc982e4f
commit 609c0e7e0d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/models/sense_voice/encoder.py
View File

@ -42,7 +42,7 @@ def sense_voice_encode_forward(
olens = None
if use_padmask and olens is not None:
padding_mask = (~make_pad_mask(olens)[:, None, :]).to(torch.bool).to(x.device)
padding_mask = (~make_pad_mask(olens)[:, :, None]).to(torch.bool).to(x.device)
else:
padding_mask = None