From 80e0e7e1a56191e95fdd938e3255dc37b4e7ef4f Mon Sep 17 00:00:00 2001 From: Bernard Maltais Date: Sun, 30 Oct 2022 13:37:42 -0400 Subject: [PATCH] Add dependencies --- README.md | 21 ++++++++++++++++----- 1 file changed, 16 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 7156fce..fce5921 100644 --- a/README.md +++ b/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