Dreambooth and Fine Tuning for Stable Diffusion 1.5 and 2 with this Versatile Script
Been playing with this for a few days now. Took a while to work out kinks, but things seem smooth now. Take a look at this project if you are trying to fine tune or dreambooth train a Stable Diffusion 1.5 or 2 model.
Github page for Kohya_SS script
https://github.com/bmaltais/kohya_ss
use the launch flag --max_train_steps = 50000 or another high value to allow the script to continue after 1 epoch
Intro https://www.youtube.com/watch?v=MX7GTcHr9oI&t=0m1s
Rambling about memory usage https://www.youtube.com/watch?v=MX7GTcHr9oI&t=1m30s
Setup https://www.youtube.com/watch?v=MX7GTcHr9oI&t=3m0s
Processing images and generating captions https://www.youtube.com/watch?v=MX7GTcHr9oI&t=6m0s
Altering captions with Notepad++ https://www.youtube.com/watch?v=MX7GTcHr9oI&t=7m30s
Dataset structure https://www.youtube.com/watch?v=MX7GTcHr9oI&t=8m15s
Launch script example and script launch options https://www.youtube.com/watch?v=MX7GTcHr9oI&t=8m50s
Comparing the fine tuned model to SD 1.5 base at various epochs
https://www.youtube.com/watch?v=MX7GTcHr9oI&t=10m30s
Bye for now, back soon with another https://www.youtube.com/watch?v=MX7GTcHr9oI&t=15m15s
Startup shell script
https://pastebin.com/wSQXxnN6
Command line task list:
https://pastebin.com/A0M7QuE3
Use the script from the Shivam Shiraro diffusers library to convert models to ckpt for the WebUI:
python convert_diffusers_to_original_stable_diffusion.py --model_path /IN/MODEL/DIR --checkpoint_path /OUT/FILE.CKPT