ai-toolkit/jobs
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
..
process Add `push_to_hub` to the trainer (#109) 2024-08-22 21:18:56 -06:00
BaseJob.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -06:00
ExtensionJob.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -06:00
ExtractJob.py Adjusted some names and allow to ignore text encoder on extract 2023-07-16 12:52:39 -06:00
GenerateJob.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -06:00
MergeJob.py Setup the base process for merging things. WIP 2023-07-20 07:39:31 -06:00
ModJob.py Added Model rescale and prepared a release upgrade 2023-08-01 13:49:54 -06:00
TrainJob.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -06:00
__init__.py Added extensions and an example extension that merges models 2023-08-04 09:37:24 -06:00