Code for reproducing some key results of our paper Speed up the inference of diffusion models via shortcut MCMC sampling (https://arxiv.org/abs/2301.01206).

jaxlib, tensorflow, etc

python global_vdm_2d.py

joinFlag = False # add the flag whether we need joined learning

freeze_x_decoder = False

shortcut = True

K=10 default, which can change based on experiments

we demonstrate our approach based on a 2D swirl dataset and using MLPs, most code base are from VDM Link to open in Colab.

if you think it is helpful please cite: Gang Chen, Speed up the inference of diffusion models via shortcut MCMC sampling (https://arxiv.org/abs/2301.01206).