ai-toolkit/jobs
PlagueKind 089e41dd1c
Compile improvements - auto cache size, fix fullgraph setting, fix triton detection (#899)
* Compile improvements - auto cache size, fix fullgraph setting, fix triton detection

* remove forced torchao no longer needed
2026-06-24 07:59:41 -06:00
..
process Compile improvements - auto cache size, fix fullgraph setting, fix triton detection (#899) 2026-06-24 07:59:41 -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
__init__.py Added extensions and an example extension that merges models 2023-08-04 09:37:24 -06:00