From 6715da78bd632ab6a4762fe543fd5e3b1124cfcf Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Thu, 19 Jan 2023 23:14:40 -0500
Subject: [PATCH] update config

---
 cli/train.py   |  8 +++++---
 config.json    |  2 +-
 ui-config.json | 33 ++++++++++++++++++++++++++++++++-
 3 files changed, 38 insertions(+), 5 deletions(-)

diff --git a/cli/train.py b/cli/train.py
index 113952e49..5170fe919 100755
--- a/cli/train.py
+++ b/cli/train.py
@@ -146,7 +146,7 @@ async def preprocess(params):
                 log.error({ 'preprocess video extract': 'no images' })
     elif os.path.isdir(params.src):
         if params.preprocess == 'builtin':
-            res = preprocess_builtin(params)
+            res = await preprocess_builtin(params)
         if params.preprocess == 'custom':
             t0 = time.perf_counter()
             args.preprocess.process_src = params.src
@@ -273,12 +273,14 @@ async def train(params):
 
     if params.grad == -1:
         args.train_embedding.gradient_step = len(imgs) // args.train_embedding.batch_size
+        divisor = args.train_embedding.gradient_step // 60
+        args.train_embedding.gradient_step = args.train_embedding.gradient_step // (1 + divisor)
         log.info({ 'dynamic gradient step': args.train_embedding.gradient_step })
         if params.steps == -1:
             args.train_embedding.steps = params.maxsteps // args.train_embedding.gradient_step
             log.info({ 'dynamic steps': args.train_embedding.steps })
-    
-    epoch_size = args.train_embedding.batch_size * args.train_embedding.gradient_step
+
+    epoch_size = args.train_embedding.batch_size * args.train_embedding.gradient_step  
     if args.train_embedding.create_image_every == -1:
         args.train_embedding.create_image_every = args.train_embedding.steps // 10
     if args.train_embedding.save_embedding_every == -1:
diff --git a/config.json b/config.json
index 9dca852f8..6d467bd88 100644
--- a/config.json
+++ b/config.json
@@ -60,7 +60,7 @@
     "training_enable_tensorboard": true,
     "training_tensorboard_save_images": false,
     "training_tensorboard_flush_every": 120,
-    "sd_model_checkpoint": "sd-v15-runwayml.ckpt",
+    "sd_model_checkpoint": "sd-v15-runwayml.ckpt [cc6cb27103]",
     "sd_checkpoint_cache": 0,
     "sd_vae_checkpoint_cache": 0,
     "sd_vae": "vae-ft-mse-840000-ema-pruned.ckpt",
diff --git a/ui-config.json b/ui-config.json
index 8cb36e20d..9375df956 100644
--- a/ui-config.json
+++ b/ui-config.json
@@ -734,5 +734,36 @@
     "txt2img/Styles/visible": false,
     "txt2img/Styles/value": [],
     "img2img/Styles/visible": false,
-    "img2img/Styles/value": []
+    "img2img/Styles/value": [],
+    "modelmerger/Bake in VAE/visible": true,
+    "modelmerger/Bake in VAE/value": "None",
+    "train/Auto-sized crop/visible": true,
+    "train/Auto-sized crop/value": false,
+    "train/Dimension lower bound/visible": true,
+    "train/Dimension lower bound/value": 384,
+    "train/Dimension lower bound/minimum": 64,
+    "train/Dimension lower bound/maximum": 2048,
+    "train/Dimension lower bound/step": 8,
+    "train/Dimension upper bound/visible": true,
+    "train/Dimension upper bound/value": 768,
+    "train/Dimension upper bound/minimum": 64,
+    "train/Dimension upper bound/maximum": 2048,
+    "train/Dimension upper bound/step": 8,
+    "train/Area lower bound/visible": true,
+    "train/Area lower bound/value": 4096,
+    "train/Area lower bound/minimum": 4096,
+    "train/Area lower bound/maximum": 4194304,
+    "train/Area lower bound/step": 1,
+    "train/Area upper bound/visible": true,
+    "train/Area upper bound/value": 409600,
+    "train/Area upper bound/minimum": 4096,
+    "train/Area upper bound/maximum": 4194304,
+    "train/Area upper bound/step": 1,
+    "train/Resizing objective/visible": true,
+    "train/Resizing objective/value": "Maximize area",
+    "train/Error threshold/visible": true,
+    "train/Error threshold/value": 0.1,
+    "train/Error threshold/minimum": 0,
+    "train/Error threshold/maximum": 1,
+    "train/Error threshold/step": 0.01
 }
\ No newline at end of file