flux2 vae