Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-06 13:00:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix training vectors

Browse Source
This commit is contained in:
Vladimir Mandic
2023-01-25 17:49:27 -05:00
parent 8dc09e9c9b
commit 8dec75d99e
4 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
automatic.sh
+1 -1
View File
@@ -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`
cli/train.py
+3 -1
View File
@@ -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")
extensions-builtin/sd-extension-steps-animation
+1 -1
Submodule extensions-builtin/sd-extension-steps-animation updated: 12f3efd13b...71006b8905
ui-config.json
+1 -1
View File
@@ -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,