Stable Diffusion Vae Training