summaryrefslogtreecommitdiffstats
Commit message (Collapse)AuthorAgeFilesLines
* Restored functional trainerVolpeon2023-01-155-104/+112
|
* Fixed Conda envVolpeon2023-01-151-2/+4
|
* UpdateVolpeon2023-01-155-162/+106
|
* Removed unused code, put training callbacks in dataclassVolpeon2023-01-157-1470/+40
|
* Added functional TI strategyVolpeon2023-01-153-78/+312
|
* Added functional trainerVolpeon2023-01-153-37/+101
|
* UpdateVolpeon2023-01-146-127/+33
|
* UpdateVolpeon2023-01-144-15/+15
|
* WIP: Modularization ("free(): invalid pointer" my ass)Volpeon2023-01-1411-286/+1541
|
* TI: Prepare UNet with Accelerate as wellVolpeon2023-01-143-39/+41
|
* FixVolpeon2023-01-144-6/+6
|
* CleanupVolpeon2023-01-147-131/+103
|
* Unified training script structureVolpeon2023-01-132-130/+84
|
* Reverted modularization mostlyVolpeon2023-01-137-613/+458
|
* More modularizationVolpeon2023-01-139-653/+677
|
* Simplified step calculationsVolpeon2023-01-132-33/+33
|
* Removed PromptProcessor, modularized training loopVolpeon2023-01-139-293/+334
|
* Added TI decay start offsetVolpeon2023-01-132-3/+9
|
* Code deduplicationVolpeon2023-01-136-146/+149
|
* UpdateVolpeon2023-01-123-34/+63
|
* Fixed TI decayVolpeon2023-01-122-9/+12
|
* Disable Adam weight decayVolpeon2023-01-121-1/+1
|
* FixVolpeon2023-01-112-5/+5
|
* HeckVolpeon2023-01-111-1/+1
|
* TI: Use grad clipping from LoRA #104Volpeon2023-01-114-12/+15
|
* Better defaultsVolpeon2023-01-102-7/+6
|
* FixVolpeon2023-01-102-2/+2
|
* Added arg to disable tag shufflingVolpeon2023-01-104-18/+37
|
* Enable buckets for validation, fixed vaildation repeat argVolpeon2023-01-093-10/+5
|
* Add --valid_set_repeatVolpeon2023-01-093-1/+37
|
* No cache after allVolpeon2023-01-081-17/+7
|
* Cache token IDs in datasetVolpeon2023-01-081-8/+20
|
* FixVolpeon2023-01-082-6/+6
|
* Improved aspect ratio bucketingVolpeon2023-01-083-4/+61
|
* Fixed aspect ratio bucketingVolpeon2023-01-081-3/+5
|
* CleanupVolpeon2023-01-081-15/+25
|
* Fixed aspect ratio bucketing; allow passing token IDs to pipelineVolpeon2023-01-085-68/+102
|
* Improved aspect ratio bucketingVolpeon2023-01-084-223/+237
|
* CleanupVolpeon2023-01-073-178/+201
|
* Made aspect ratio bucketing configurableVolpeon2023-01-073-18/+37
|
* Added progressive aspect ratio bucketingVolpeon2023-01-075-126/+151
|
* UpdateVolpeon2023-01-073-11/+6
|
* FixVolpeon2023-01-061-2/+4
|
* Relaxed ema_power for TIVolpeon2023-01-061-1/+1
|
* UpdateVolpeon2023-01-064-33/+26
|
* Use context manager for EMA, on_train/eval hooksVolpeon2023-01-063-81/+92
|
* Package updateVolpeon2023-01-061-1/+1
|
* Add prompt template argument to inferenceVolpeon2023-01-061-0/+7
|
* Add contextmanager to EMAModel to apply weights temporarilyVolpeon2023-01-062-33/+36
|
* Log EMA decayVolpeon2023-01-051-1/+2
|