from diffusers import AutoencoderKL

vae = AutoencoderKL.from_pretrained("AiArtLab/sdxl_vae").cuda().half()