diff --git a/README.md b/README.md index 9149afc..ea38501 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,6 @@ # HOWTO -https://note.com/kohya_ss/n/nee3ed1649fb6 +This repo provide all the required config to run the Dreambooth version found in this note: https://note.com/kohya_ss/n/nee3ed1649fb6 ## Required Dependencies @@ -42,13 +42,35 @@ accelerate config: - fp16 ``` +## Folders configuration + +Refer to the note to understand how to create the folde structure. In short it should look like: + +``` + +|- reg_ + |- _ +|- train_ + |- _ +``` + +Example for `sks dog` + +``` +my_sks_dog_dreambooth +|- reg_dog + |- 1_sks dog +|- train_dog + |- 20_sks dog +``` + ## 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_bernard\train_man" ` - --reg_data_dir="D:\dreambooth\train_bernard\reg_man" ` + --train_data_dir="D:\dreambooth\train_bernard\reg_man" ` --output_dir="D:\dreambooth\train_bernard" ` --prior_loss_weight=1.0 ` --resolution=512 `