Modify the architecture for multi-token prediction as in https://arxiv.org/abs/2404.19737.
Modify the architecture for multi-token prediction as in https://arxiv.org/abs/2404.19737.