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

onnx

Browse Source
This commit is contained in:
游雁 2023-04-07 14:11:55 +08:00
parent eb82674d88
commit 9f6445d39b
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/export/models/vad_realtime_transformer.py
View File

@ -66,7 +66,7 @@ class VadRealtimeTransformer(nn.Module):
length = 10 length = 10
text_indexes = torch.tensor([[266757, 266757, 266757, 266757, 266757, 266757, 266757, 266757, 266757, 266757]], dtype=torch.int32) text_indexes = torch.tensor([[266757, 266757, 266757, 266757, 266757, 266757, 266757, 266757, 266757, 266757]], dtype=torch.int32)
text_lengths = torch.tensor([length], dtype=torch.int32) text_lengths = torch.tensor([length], dtype=torch.int32)
vad_masks = vad_mask(10, 3, dtype=torch.float32)[None, None, :, :] vad_masks = vad_mask(10, 2, dtype=torch.float32)[None, None, :, :]
sub_masks = torch.ones(length, length, dtype=torch.float32) sub_masks = torch.ones(length, length, dtype=torch.float32)
sub_masks = torch.tril(sub_masks).type(torch.float32) sub_masks = torch.tril(sub_masks).type(torch.float32)
return (text_indexes, text_lengths, vad_masks, sub_masks[None, None, :, :]) return (text_indexes, text_lengths, vad_masks, sub_masks[None, None, :, :])