Conditioning By Replacement in Diffusion Models

Training

Main config files in algorithms/diffusion_forcing/models/diffusion.py and algorithms/diffusion_forcing/df_base.py.

Credit

If you find this research template helpful to you, please cite us as

@misc{chen2024diffusionforcingnexttokenprediction,
      title={Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion},
      author={Boyuan Chen and Diego Marti Monso and Yilun Du and Max Simchowitz and Russ Tedrake and Vincent Sitzmann},
      year={2024},
      eprint={2407.01392},
      archivePrefix={arXiv},
      primaryClass={cs.LG},
      url={https://arxiv.org/abs/2407.01392},
}