whisper_decoder: Text Decoder
Description
Full Whisper decoder: token embedding + positional embedding + transformer layers.
Usage
whisper_decoder(n_vocab, n_ctx, n_state, n_head, n_layer)
Arguments
- n_vocab
Vocabulary size
- n_ctx
Maximum context length
- n_state
Hidden dimension
- n_head
Number of attention heads
- n_layer
Number of transformer layers