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:
游雁 2024-07-04 23:33:51 +08:00
parent e969be589e
commit 496ca8eddb
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/llm_asr/model.py
View File

@ -2754,7 +2754,7 @@ class LLMASR5(nn.Module):
dtype=torch.int32,
).to(inputs_embeds.device)
for i in range(token_num):
hidden_states_out[0, i, :] = hidden_states[1][-1][0, 0, :].to(torch.float32)
hidden_states_out[0, i, :] = hidden_states[i][-1][0, 0, :].to(torch.float32)
speech_tokens = self.audio_decode(hidden_states_out, hidden_states_out_len)[
:, :, 0