Choose tokens
×no parameters
At every position take the token with the largest logit.
- in
- [B batch, T positions, 50259 vocabulary]
- out
- [B batch, T positions]
Explanation
Nothing is sampled here: all the randomness of generation is in the latents. Every position is decided at once, from one pass of the decoders.
Sources1
Direct evidence
Generation: the sampled latents pass the latent decoder and the token decoder, each position takes the token of the largest logit, and the tokens are decoded to text, special tokens kept unless truncation is on.
The implementation, pinned at revision e18c42acdfb7, not yet public