summaryrefslogtreecommitdiffstats
path: root/training
Commit message (Expand)AuthorAgeFilesLines
* Add LoraVolpeon2023-02-074-37/+214
* Restored LR finderVolpeon2023-01-206-393/+82
* Move Accelerator preparation into strategyVolpeon2023-01-193-16/+48
* UpdateVolpeon2023-01-174-14/+19
* FixVolpeon2023-01-171-4/+5
* FixVolpeon2023-01-171-1/+0
* Make embedding decay work like Adam decayVolpeon2023-01-171-9/+5
* UpdateVolpeon2023-01-172-8/+8
* UpdateVolpeon2023-01-174-21/+38
* Training updateVolpeon2023-01-163-12/+15
* Moved multi-TI code from Dreambooth to TI scriptVolpeon2023-01-161-3/+14
* More training adjustmentsVolpeon2023-01-163-8/+9
* Handle empty validation datasetVolpeon2023-01-163-47/+58
* Extended Dreambooth: Train TI tokens separatelyVolpeon2023-01-161-0/+1
* Implemented extended Dreambooth trainingVolpeon2023-01-161-3/+4
* Added Dreambooth strategyVolpeon2023-01-151-0/+183
* Restored functional trainerVolpeon2023-01-152-27/+83
* UpdateVolpeon2023-01-153-119/+64
* Removed unused code, put training callbacks in dataclassVolpeon2023-01-152-44/+39
* Added functional TI strategyVolpeon2023-01-152-0/+282
* Added functional trainerVolpeon2023-01-151-1/+74
* UpdateVolpeon2023-01-142-122/+24
* UpdateVolpeon2023-01-141-10/+9
* WIP: Modularization ("free(): invalid pointer" my ass)Volpeon2023-01-143-220/+130
* TI: Prepare UNet with Accelerate as wellVolpeon2023-01-142-27/+26
* FixVolpeon2023-01-141-1/+1
* CleanupVolpeon2023-01-142-60/+63
* Reverted modularization mostlyVolpeon2023-01-135-531/+70
* More modularizationVolpeon2023-01-136-32/+541
* Simplified step calculationsVolpeon2023-01-131-20/+22
* Removed PromptProcessor, modularized training loopVolpeon2023-01-132-10/+208
* Code deduplicationVolpeon2023-01-131-0/+55
* UpdateVolpeon2023-01-121-4/+7
* FixVolpeon2023-01-111-2/+2
* TI: Use grad clipping from LoRA #104Volpeon2023-01-111-1/+1
* Added arg to disable tag shufflingVolpeon2023-01-101-10/+10
* Fixed aspect ratio bucketing; allow passing token IDs to pipelineVolpeon2023-01-081-6/+8
* Improved aspect ratio bucketingVolpeon2023-01-081-1/+1
* CleanupVolpeon2023-01-071-0/+54
* Made aspect ratio bucketing configurableVolpeon2023-01-071-7/+2
* Added progressive aspect ratio bucketingVolpeon2023-01-071-2/+2
* UpdateVolpeon2023-01-061-6/+6
* Use context manager for EMA, on_train/eval hooksVolpeon2023-01-062-27/+26
* Add contextmanager to EMAModel to apply weights temporarilyVolpeon2023-01-061-0/+12
* Added EMA to TIVolpeon2023-01-051-5/+95
* Fix LR finderVolpeon2023-01-051-7/+23
* UpdateVolpeon2023-01-051-1/+4
* Various cleanupsVolpeon2023-01-051-0/+75
* UpdateVolpeon2023-01-041-1/+6
* Various updatesVolpeon2023-01-041-4/+0