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
|
||||
|
||||
|
||||
## 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
|
||||
|
Loading…
Reference in New Issue
Block a user