From 8dec75d99e86b28274db2bcfe8315f525d8f1fbb Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Wed, 25 Jan 2023 17:49:27 -0500 Subject: [PATCH] fix training vectors --- automatic.sh | 2 +- cli/train.py | 4 +++- extensions-builtin/sd-extension-steps-animation | 2 +- ui-config.json | 2 +- 4 files changed, 6 insertions(+), 4 deletions(-) diff --git a/automatic.sh b/automatic.sh index aa6494400..9a076f13f 100755 --- a/automatic.sh +++ b/automatic.sh @@ -8,7 +8,7 @@ PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.9,max_split_size_mb:512 CUDA_LAUNCH_BLOCKING=0 CUDA_CACHE_DISABLE=0 CUDA_AUTO_BOOST=1 -CUDA_DEVICE_DEFAULT_PERSISTING_L2_CACHE_PERCENTAGE_LIMIT=100 +CUDA_DEVICE_DEFAULT_PERSISTING_L2_CACHE_PERCENTAGE_LIMIT=0 if [ "$PYTHON" == "" ]; then PYTHON=`which python` diff --git a/cli/train.py b/cli/train.py index f2fb2fb96..fc53f3f2b 100755 --- a/cli/train.py +++ b/cli/train.py @@ -252,6 +252,8 @@ async def create(params): vectors = 4 else: vectors = 6 + else: + vectors = params.vectors if os.path.exists(params.name) and os.path.isfile(params.name): log.info({ 'deleting existing embedding': { 'name': params.name } }) os.remove(params.name) @@ -426,7 +428,7 @@ async def main(): parser.add_argument("--init", type = str, default = "person", required = False, help = "initialization class, default: %(default)s") parser.add_argument("--dst", type = str, default = "/tmp", required = False, help = "destination image folder for processed images, default: %(default)s") parser.add_argument("--steps", type = int, default = -1, required = False, help = "training steps, default: %(default)s") - parser.add_argument("--maxsteps", type = int, default = 4000, required = False, help = "max training steps used when dynamic gradient is active, default: %(default)s") + parser.add_argument("--maxsteps", type = int, default = 5000, required = False, help = "max training steps used when dynamic gradient is active, default: %(default)s") parser.add_argument("--vectors", type = int, default = -1, required = False, help = "number of vectors per token, default: dynamic based on number of input images") parser.add_argument("--batch", type = int, default = 1, required = False, help = "batch size, default: %(default)s") parser.add_argument("--rate", type = str, default = "", required = False, help = "learn rate, default: dynamic") diff --git a/extensions-builtin/sd-extension-steps-animation b/extensions-builtin/sd-extension-steps-animation index 12f3efd13..71006b890 160000 --- a/extensions-builtin/sd-extension-steps-animation +++ b/extensions-builtin/sd-extension-steps-animation @@ -1 +1 @@ -Subproject commit 12f3efd13b25790bbf9dc8dfdb3c276e7e7902d1 +Subproject commit 71006b8905e881636f457ed702d3a65e4b334bfe diff --git a/ui-config.json b/ui-config.json index 8ca3dac2a..7bd66e6c4 100644 --- a/ui-config.json +++ b/ui-config.json @@ -369,7 +369,7 @@ "customscript/save_steps_animation.py/txt2img/Duration/maximum": 120, "customscript/save_steps_animation.py/txt2img/Duration/step": 0.1, "customscript/save_steps_animation.py/txt2img/Skip steps/visible": true, - "customscript/save_steps_animation.py/txt2img/Skip steps/value": 5, + "customscript/save_steps_animation.py/txt2img/Skip steps/value": 0, "customscript/save_steps_animation.py/txt2img/Skip steps/minimum": 0, "customscript/save_steps_animation.py/txt2img/Skip steps/maximum": 100, "customscript/save_steps_animation.py/txt2img/Skip steps/step": 1,