This repository provides a Windows-focused Gradio GUI for Kohya's Stable Diffusion trainers
https://github.com/bmaltais/kohya_ss
bitsandbytes_windows | ||
.gitignore | ||
README.md | ||
requirements.txt | ||
train_db_fixed_v4.py | ||
train_db_fixed_v6.py |
HOWTO
https://note.com/kohya_ss/n/nee3ed1649fb6
python -m venv --system-site-packages venv
.\venv\Scripts\activate
11.6:
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
pip install --upgrade diffusers
pip install -r requirements.txt
pip install OmegaConf
pip install pytorch_lightning
pip install -U -I --no-deps https://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/f/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl
# Setup bitsandbytes wioth Adam8bit support for windows: https://note.com/kohya_ss/n/n47f654dc161e
pip install bitsandbytes==0.35.0
cp .\bitsandbytes_windows\*.dll .\venv\Lib\site-packages\bitsandbytes\
cp .\bitsandbytes_windows\cextension.py .\venv\Lib\site-packages\bitsandbytes\cextension.py
cp .\bitsandbytes_windows\main.py .\venv\Lib\site-packages\bitsandbytes\cuda_setup\main.py
accelerate config
accelerate launch --num_cpu_threads_per_process 6 train_db_fixed_v6.py `
--pretrained_model_name_or_path="d:\models\v1-5-pruned.ckpt" `
--train_data_dir="D:\dreambooth\train_artstyle" `
--reg_data_dir="D:\dreambooth\reg_artstyle" `
--output_dir="D:\dreambooth" `
--prior_loss_weight=1.0 `
--resolution=512 `
--train_batch_size=1 `
--learning_rate=1e-6 `
--max_train_steps=1600 `
--use_8bit_adam `
--xformers `
--mixed_precision="fp16" `
--cache_latents `
--gradient_checkpointing `
--save_every_n_epochs=500