Add dependencies

This commit is contained in:
Bernard Maltais 2022-10-30 13:37:42 -04:00
parent 1b3ad40471
commit 80e0e7e1a5

View File

@ -3,6 +3,13 @@
https://note.com/kohya_ss/n/nee3ed1649fb6
## Required Dependencies
Python 3.10.6 and Git:
Windows: download and run installers for Python 3.10.6 (https://www.python.org/ftp/python/3.10.6/python-3.10.6-amd64.exe) and git (https://git-scm.com/download/win)
## Installation
```
git clone https://github.com/bmaltais/kohya_ss.git
cd kohya_ss
@ -31,20 +38,24 @@ accelerate config:
- NO
- All
- fp16
```
## Execution
```
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" `
--train_data_dir="D:\dreambooth\train_bernard\train_man" `
--reg_data_dir="D:\dreambooth\train_bernard\reg_man" `
--output_dir="D:\dreambooth\train_bernard" `
--prior_loss_weight=1.0 `
--resolution=512 `
--train_batch_size=1 `
--learning_rate=1e-6 `
--max_train_steps=1600 `
--max_train_steps=2100 `
--use_8bit_adam `
--xformers `
--mixed_precision="fp16" `
--cache_latents `
--gradient_checkpointing `
--save_every_n_epochs=500
--save_every_n_epochs=1