Train VAE models using Diffusers, Datasets, Accelerate libraries - View it on GitHub