The model already has included vae, there is no need to add additional vae