Add dependencies
This commit is contained in:
parent
1b3ad40471
commit
80e0e7e1a5
21
README.md
21
README.md
@ -3,6 +3,13 @@
|
|||||||
https://note.com/kohya_ss/n/nee3ed1649fb6
|
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
|
git clone https://github.com/bmaltais/kohya_ss.git
|
||||||
cd kohya_ss
|
cd kohya_ss
|
||||||
@ -31,20 +38,24 @@ accelerate config:
|
|||||||
- NO
|
- NO
|
||||||
- All
|
- All
|
||||||
- fp16
|
- fp16
|
||||||
|
```
|
||||||
|
|
||||||
|
## Execution
|
||||||
|
|
||||||
|
```
|
||||||
accelerate launch --num_cpu_threads_per_process 6 train_db_fixed_v6.py `
|
accelerate launch --num_cpu_threads_per_process 6 train_db_fixed_v6.py `
|
||||||
--pretrained_model_name_or_path="d:\models\v1-5-pruned.ckpt" `
|
--pretrained_model_name_or_path="d:\models\v1-5-pruned.ckpt" `
|
||||||
--train_data_dir="D:\dreambooth\train_artstyle" `
|
--train_data_dir="D:\dreambooth\train_bernard\train_man" `
|
||||||
--reg_data_dir="D:\dreambooth\reg_artstyle" `
|
--reg_data_dir="D:\dreambooth\train_bernard\reg_man" `
|
||||||
--output_dir="D:\dreambooth" `
|
--output_dir="D:\dreambooth\train_bernard" `
|
||||||
--prior_loss_weight=1.0 `
|
--prior_loss_weight=1.0 `
|
||||||
--resolution=512 `
|
--resolution=512 `
|
||||||
--train_batch_size=1 `
|
--train_batch_size=1 `
|
||||||
--learning_rate=1e-6 `
|
--learning_rate=1e-6 `
|
||||||
--max_train_steps=1600 `
|
--max_train_steps=2100 `
|
||||||
--use_8bit_adam `
|
--use_8bit_adam `
|
||||||
--xformers `
|
--xformers `
|
||||||
--mixed_precision="fp16" `
|
--mixed_precision="fp16" `
|
||||||
--cache_latents `
|
--cache_latents `
|
||||||
--gradient_checkpointing `
|
--gradient_checkpointing `
|
||||||
--save_every_n_epochs=500
|
--save_every_n_epochs=1
|
||||||
|
Loading…
Reference in New Issue
Block a user