sd-scripts icon indicating copy to clipboard operation
sd-scripts copied to clipboard
verified publisher icon kohya-ss

Train VAE?

Open dill-shower opened this issue 2 years ago • 1 comments

How to train VAE for model? Fine_tune.py train unet only network

dill-shower avatar Apr 07 '23 16:04 dill-shower

A recently developed project for this purpose is https://github.com/lavinal712/AutoencoderKL, which provides scripts for specifically training the VAE for SD15, SDXL, and FLUX based models. Alternative, older, but similarly interesting training scripts can be found here:

  1. https://github.com/kukaiN/vae_finetune

lzardy avatar Jun 20 '25 04:06 lzardy