Jaret Burkett 5365200da1 Added ability to add models to finetune as plugins. Also added flux2 new arch via that method.

2025-03-27 16:07:00 -06:00

..

get_model.py

Added ability to add models to finetune as plugins. Also added flux2 new arch via that method.

2025-03-27 16:07:00 -06:00

inverse_cfg.py

Small tweaks and fixes for specialized ip adapter training

2024-03-26 11:35:26 -06:00

mask.py

Added an inpainting mask generator for training inpainting if inpaint mask is not provided

2025-03-25 12:16:10 -06:00

quantize.py

Added ability to quantize with torchao

2025-03-20 16:28:54 -06:00

wavelet_loss.py

Added experimental wavelet loss

2025-03-26 18:11:23 -06:00