OSSForks/whisper
1
0
Fork 0
mirror of https://github.com/zebrajr/whisper.git synced 2025-12-06 00:19:50 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix doc of TextDecoder (#1526)

Browse Source 
Signed-off-by: haoshengqiang <haoshengqiang@xiaohongshu.com>
Co-authored-by: haoshengqiang <haoshengqiang@xiaohongshu.com>
This commit is contained in:
sqhao 2023-09-19 07:09:59 +08:00 committed by GitHub
parent 29b7df6231
commit 21010ef454
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
whisper/model.py
View File

@ -197,7 +197,7 @@ class TextDecoder(nn.Module):
"""
x : torch.LongTensor, shape = (batch_size, <= n_ctx)
the text tokens
xa : torch.Tensor, shape = (batch_size, n_mels, n_audio_ctx)
xa : torch.Tensor, shape = (batch_size, n_audio_ctx, n_audio_state)
the encoded audio features to be attended on
"""
offset = next(iter(kv_cache.values())).shape[1] if kv_cache else 0