Jaret Burkett 3b1f7b0948 Allow user to set the attention backend. Add method to recomver from the occasional OOM if it is a rare event. Still exit if it ooms 3 times in a row.

2025-09-27 08:56:15 -06:00

..

process

Allow user to set the attention backend. Add method to recomver from the occasional OOM if it is a rare event. Still exit if it ooms 3 times in a row.

2025-09-27 08:56:15 -06:00

__init__.py

Added extensions and an example extension that merges models

2023-08-04 09:37:24 -06:00

BaseJob.py

Added a basic torch profiler that can be used in config during development to find some obvious issues.

2025-06-17 13:03:39 -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

Removed all submodules. Submodule free now, yay.

2025-04-18 10:39:15 -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

Removed all submodules. Submodule free now, yay.

2025-04-18 10:39:15 -06:00