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