code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion" - View it on GitHub