summaryrefslogtreecommitdiffstats
path: root/train_ti.py
Commit message (Expand)AuthorAgeFilesLines
* Add LoraVolpeon2023-02-071-4/+6
* Restored LR finderVolpeon2023-01-201-2/+19
* Move Accelerator preparation into strategyVolpeon2023-01-191-3/+3
* Smaller emb decayVolpeon2023-01-171-1/+1
* Make embedding decay work like Adam decayVolpeon2023-01-171-12/+4
* UpdateVolpeon2023-01-171-49/+64
* Training updateVolpeon2023-01-161-3/+5
* If valid set size is 0, re-use one image from train setVolpeon2023-01-161-5/+1
* Moved multi-TI code from Dreambooth to TI scriptVolpeon2023-01-161-107/+114
* More training adjustmentsVolpeon2023-01-161-5/+12
* Handle empty validation datasetVolpeon2023-01-161-6/+3
* Implemented extended Dreambooth trainingVolpeon2023-01-161-40/+22
* Added Dreambooth strategyVolpeon2023-01-151-23/+23
* Restored functional trainerVolpeon2023-01-151-61/+21
* UpdateVolpeon2023-01-151-36/+38
* Removed unused code, put training callbacks in dataclassVolpeon2023-01-151-48/+1
* Added functional TI strategyVolpeon2023-01-151-78/+30
* Added functional trainerVolpeon2023-01-151-26/+23
* UpdateVolpeon2023-01-141-5/+5
* UpdateVolpeon2023-01-141-3/+4
* WIP: Modularization ("free(): invalid pointer" my ass)Volpeon2023-01-141-59/+15
* TI: Prepare UNet with Accelerate as wellVolpeon2023-01-141-12/+15
* FixVolpeon2023-01-141-2/+2
* CleanupVolpeon2023-01-141-21/+12
* Unified training script structureVolpeon2023-01-131-3/+6
* Reverted modularization mostlyVolpeon2023-01-131-81/+386
* More modularizationVolpeon2023-01-131-409/+70
* Simplified step calculationsVolpeon2023-01-131-13/+11
* Removed PromptProcessor, modularized training loopVolpeon2023-01-131-215/+53
* Added TI decay start offsetVolpeon2023-01-131-2/+8
* Code deduplicationVolpeon2023-01-131-60/+26
* UpdateVolpeon2023-01-121-16/+14
* Fixed TI decayVolpeon2023-01-121-8/+3
* Disable Adam weight decayVolpeon2023-01-121-1/+1
* FixVolpeon2023-01-111-3/+3
* HeckVolpeon2023-01-111-1/+1
* TI: Use grad clipping from LoRA #104Volpeon2023-01-111-8/+11
* Better defaultsVolpeon2023-01-101-4/+4
* FixVolpeon2023-01-101-1/+1
* Added arg to disable tag shufflingVolpeon2023-01-101-6/+16
* Enable buckets for validation, fixed vaildation repeat argVolpeon2023-01-091-4/+1
* Add --valid_set_repeatVolpeon2023-01-091-0/+22
* FixVolpeon2023-01-081-1/+1
* Improved aspect ratio bucketingVolpeon2023-01-081-1/+15
* Fixed aspect ratio bucketing; allow passing token IDs to pipelineVolpeon2023-01-081-11/+13
* Improved aspect ratio bucketingVolpeon2023-01-081-48/+37
* CleanupVolpeon2023-01-071-44/+16
* Made aspect ratio bucketing configurableVolpeon2023-01-071-0/+13
* Added progressive aspect ratio bucketingVolpeon2023-01-071-48/+46
* UpdateVolpeon2023-01-071-3/+3