Jaret Burkett ce4c5291a0 Added experimental wavelet loss

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

..

get_model.py

Added cogview4. Loss still needs work.

2025-03-04 18:43:52 -07: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