ai-toolkit/jobs/process
Jaret Burkett d35733ac06 Added support for training ssd-1B. Added support for saving models into diffusers format. We can currently save in safetensors format for ssd-1b, but diffusers cannot load it yet. 2023-11-03 05:01:16 -06:00
..
models Fixes to esrgan trainer. Moved logic for sd prompt embeddings out of diffusers pipeline so I can manipulate it 2023-09-16 17:41:07 -06:00
BaseExtensionProcess.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -06:00
BaseExtractProcess.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -06:00
BaseMergeProcess.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -06:00
BaseProcess.py Massive speed increases and ram optimizations 2023-10-10 06:07:55 -06:00
BaseSDTrainProcess.py Added support for training ssd-1B. Added support for saving models into diffusers format. We can currently save in safetensors format for ssd-1b, but diffusers cannot load it yet. 2023-11-03 05:01:16 -06:00
BaseTrainProcess.py Bug fixes, speed improvements, compatability adjustments withdiffusers updates 2023-09-13 07:03:53 -06:00
ExtractLoconProcess.py Adjusted some names and allow to ignore text encoder on extract 2023-07-16 12:52:39 -06:00
ExtractLoraProcess.py Added support for 3cleir, not fully tested 2023-07-16 15:35:14 -06:00
GenerateProcess.py reworked samplers. Trying to find what is wrong with diffusers sampling is sdxl 2023-09-03 07:56:09 -06:00
MergeLoconProcess.py Setup the base process for merging things. WIP 2023-07-20 07:39:31 -06:00
ModRescaleLoraProcess.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -06:00
TrainESRGANProcess.py Added some further extendability for plugins 2023-09-19 05:41:44 -06:00
TrainFineTuneProcess.py Setup base for training jobs. Added sd-scripts as a submodule 2023-07-08 13:50:59 -06:00
TrainSDRescaleProcess.py Added inbuild plugins and made one for image referenced. WIP 2023-08-10 16:02:44 -06:00
TrainSliderProcess.py Added masking to slider training. Something is still weird though 2023-11-01 14:51:29 -06:00
TrainSliderProcessOld.py Added inbuild plugins and made one for image referenced. WIP 2023-08-10 16:02:44 -06:00
TrainVAEProcess.py Added training for a custom version of ERSGAN arcitecture. Testing training now 2023-08-07 18:04:23 -06:00
__init__.py Hude rework to move the batch to a DTO to make it far more modular to the future ui 2023-08-29 10:22:19 -06:00