summaryrefslogtreecommitdiffstats
path: root/train_dreambooth.py
Commit message (Collapse)AuthorAgeFilesLines
* Added Lion optimizerVolpeon2023-02-171-11/+27
|
* Back to xformersVolpeon2023-02-171-2/+2
|
* Remove xformers, switch to Pytorch NightlyVolpeon2023-02-171-2/+2
|
* Integrated WIP UniPC schedulerVolpeon2023-02-161-1/+2
|
* UpdateVolpeon2023-02-131-5/+5
|
* Integrate Self-Attention-Guided (SAG) Stable Diffusion in my custom pipelineVolpeon2023-02-081-1/+1
|
* Fixed Lora trainingVolpeon2023-02-081-6/+6
|
* Add LoraVolpeon2023-02-071-45/+1
|
* Restored LR finderVolpeon2023-01-201-1/+9
|
* UpdateVolpeon2023-01-171-8/+6
|
* UpdateVolpeon2023-01-171-3/+2
|
* Training updateVolpeon2023-01-161-1/+5
|
* Moved multi-TI code from Dreambooth to TI scriptVolpeon2023-01-161-133/+2
|
* More training adjustmentsVolpeon2023-01-161-12/+59
|
* Pad dataset if len(items) < batch_sizeVolpeon2023-01-161-20/+20
|
* Handle empty validation datasetVolpeon2023-01-161-3/+3
|
* Extended Dreambooth: Train TI tokens separatelyVolpeon2023-01-161-71/+76
|
* Implemented extended Dreambooth trainingVolpeon2023-01-161-329/+155
|
* WIP: Modularization ("free(): invalid pointer" my ass)Volpeon2023-01-141-2/+1
|
* FixVolpeon2023-01-141-2/+2
|
* CleanupVolpeon2023-01-141-21/+12
|
* Unified training script structureVolpeon2023-01-131-127/+78
|
* Reverted modularization mostlyVolpeon2023-01-131-1/+2
|
* More modularizationVolpeon2023-01-131-207/+65
|
* Removed PromptProcessor, modularized training loopVolpeon2023-01-131-5/+2
|
* Added TI decay start offsetVolpeon2023-01-131-1/+1
|
* Code deduplicationVolpeon2023-01-131-56/+15
|
* UpdateVolpeon2023-01-121-14/+42
|
* TI: Use grad clipping from LoRA #104Volpeon2023-01-111-2/+2
|
* Better defaultsVolpeon2023-01-101-3/+2
|
* FixVolpeon2023-01-101-1/+1
|
* Added arg to disable tag shufflingVolpeon2023-01-101-1/+8
|
* Enable buckets for validation, fixed vaildation repeat argVolpeon2023-01-091-4/+1
|
* Add --valid_set_repeatVolpeon2023-01-091-0/+10
|
* Improved aspect ratio bucketingVolpeon2023-01-081-0/+27
|
* Fixed aspect ratio bucketing; allow passing token IDs to pipelineVolpeon2023-01-081-6/+8
|
* Improved aspect ratio bucketingVolpeon2023-01-081-55/+45
|
* CleanupVolpeon2023-01-071-134/+131
|
* Added progressive aspect ratio bucketingVolpeon2023-01-071-9/+3
|
* UpdateVolpeon2023-01-071-3/+3
|
* Added EMA to TIVolpeon2023-01-051-1/+1
|
* UpdateVolpeon2023-01-051-7/+7
|
* FixVolpeon2023-01-051-2/+2
|
* Various cleanupsVolpeon2023-01-051-65/+21
|
* Various updatesVolpeon2023-01-041-0/+8
|
* Better eval generatorVolpeon2023-01-041-4/+4
|
* Fixed reproducibility, more consistant validationVolpeon2023-01-041-5/+19
|
* Don't use vector_dropout by defaultVolpeon2023-01-031-1/+1
|
* Added vector dropoutVolpeon2023-01-031-5/+19
|
* Fixed LR finderVolpeon2023-01-021-2/+2
|