LazyDiT

Official repo for the AAAI 2025 paper LazyDiT.

Quick Start

Prepare the ImageNet and download pretrained DiT checkpoints from original DiT.

Go to lazydit/lazy_learning/.

Set ImageNet path and pretrained model checkpoint path in run-lazy-learning-imagenet.sh.

Then run by sh run-lazy-learning-imagenet.sh.

Citation

@inproceedings{shen2024lazydit,
    title={Lazydit: Lazy learning for the acceleration of diffusion transformers},
    author={Shen, Xuan and Song, Zhao and Zhou, Yufa and Chen, Bo and Li, Yanyu and Gong, Yifan and Zhang, Kai and Tan, Hao and Kuen, Jason and Ding, Henghui and others},
    booktitle = {AAAI},
    year      = {2025},
}