Pacific-Prime/diffusion-vae at main