ai-toolkit/jobs/process
apolinário 4d35a29c97
Add `push_to_hub` to the trainer (#109)
* add push_to_hub

* fix indentation

* indent again

* model_config

* allow samples to not exist

* repo creation fix

* dont show empty [] if widget doesnt exist

* dont submit the config and optimizer

* Unsafe to have tokens saved in the yaml file

* make sure to catch only the latest samples

* change name to slug

* formatting

* formatting

---------

Co-authored-by: multimodalart <joaopaulo.passos+multimodal@gmail.com>
2024-08-22 21:18:56 -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 Add `push_to_hub` to the trainer (#109) 2024-08-22 21:18:56 -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 New image generation img2img. various tweaks and fixes 2024-07-24 04:13:41 -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 Tons of bug fixes and improvements to special training. Fixed slider training. 2023-12-09 16:38:10 -07:00
TrainSliderProcessOld.py Added inbuild plugins and made one for image referenced. WIP 2023-08-10 16:02:44 -06:00
TrainVAEProcess.py Bug fixes and little improvements here and there. 2024-06-08 06:24:20 -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