AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16).to("cuda")                                                                                                                                        
Now we can set it to the vae of the pipeline to use it.                                                                                                                                                                