Was advised to use a different vae, it looks nice.