download checkpoint of VAE to put here
