Setting a custom VAE

It's super easy to load custom VAE's with Auto 1111 SDK. You can find a list of VAE's to choose from Civit AI here: https://civitai.com/models. To load a custom VAE in Auto 1111 SDK:

  1. Initialize your pipeline

from auto1111sdk import StableDiffusionPipeline

pipe = StableDiffusionPipeline("PATH TO YOUR MODEL SAFETENSORS FILE")
  1. Set the VAE

pipe.set_vae("PATH TO YOUR VAE SAFETENSORS FILE") # path to your vae safetensors file

And that's it! Once you change the VAE as per your specification, you shall see the changes.

Last updated