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 asr_infer.py (#622)

Browse Source 
bug fix
This commit is contained in:
aky15 2023-06-12 21:41:41 +08:00 committed by GitHub
parent d7e3ae17c3
commit 68ae83684f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/bin/asr_infer.py
View File

@ -493,9 +493,9 @@ class Speech2TextParaformer:
else:
if pre_token_length[i] == 0:
yseq = torch.tensor(
[self.asr_model.sos] + [self.asr_model.eos], device=yseq.device
[self.asr_model.sos] + [self.asr_model.eos], device=pre_acoustic_embeds.device
)
score = torch.tensor(0.0, device=yseq.device)
score = torch.tensor(0.0, device=pre_acoustic_embeds.device)
else:
yseq = am_scores.argmax(dim=-1)
score = am_scores.max(dim=-1)[0]