diff-sampler is an open-source toolbox for fast sampling of diffusion models, to provide a fair comparison of existing approaches and help researchers to develp better approaches. diff-sampler contains various model implementations, numerical-based solvers, time schedules, and other features.
This repository also includes (or will include) the official implementations of our following works:
./src/dataset_name
. Some errors and typos are fixed. Detailed running scripts are provided in launch.sh
, where we also add new scripts for evaluation of CLIP score for Stable Diffusion.If you find this repository useful, please consider citing the following paper (reverse chronological order):
@article{chen2024trajectory,
title={On the Trajectory Regularity of ODE-based Diffusion Sampling},
author={Chen, Defang and Zhou, Zhenyu and Wang, Can and Shen, Chunhua and Lyu, Siwei},
journal={arXiv preprint arXiv:2405.11326},
year={2024}
}
@article{zhou2023fast,
title={Fast ODE-based Sampling for Diffusion Models in Around 5 Steps},
author={Zhou, Zhenyu and Chen, Defang and Wang, Can and Chen, Chun},
journal={arXiv preprint arXiv:2312.00094},
year={2023}
}
@article{chen2023geometric,
title={A geometric perspective on diffusion models},
author={Chen, Defang and Zhou, Zhenyu and Mei, Jian-Ping and Shen, Chunhua and Chen, Chun and Wang, Can},
journal={arXiv preprint arXiv:2305.19947},
year={2023}
}