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

bugfix

Browse Source
This commit is contained in:
游雁 2024-05-15 16:40:12 +08:00
parent 2f7dcbad90
commit c7bfd1fdc5
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/models/sense_voice/whisper_lib/decoding.py
View File

@ -63,8 +63,8 @@ def detect_language(
else: else:
x = x.to(mel.device) x = x.to(mel.device)
# FIX(funasr): sense vocie # FIX(funasr): sense vocie
# logits = model.logits(x[:, :-1], mel)[:, -1] logits = model.logits(x[:, :-1], mel)[:, -1]
logits = model.logits(x[:, :], mel)[:, -1] # logits = model.logits(x[:, :], mel)[:, -1]
# collect detected languages; suppress all non-language tokens # collect detected languages; suppress all non-language tokens
mask = torch.ones(logits.shape[-1], dtype=torch.bool) mask = torch.ones(logits.shape[-1], dtype=torch.bool)