ai-toolkit/toolkit/util
Jaret Burkett d870e9b68a Fix backwards compatability for older versions of triton 2026-08-04 21:07:57 -06:00
..
blended_blur_noise.py Various experiments and minor bug fixes for edge cases 2025-04-25 13:44:38 -06:00
comfy_quant_import.py Fix issue with layer offloading with MinMax H3 2026-08-03 11:32:36 -06:00
convrot_quant.py Fix backwards compatability for older versions of triton 2026-08-04 21:07:57 -06:00
get_model.py Added initial support for finetuning wan i2v WIP 2025-04-07 20:34:38 -06:00
inverse_cfg.py Small tweaks and fixes for specialized ip adapter training 2024-03-26 11:35:26 -06:00
ip_adapter_utils.py Remove ip adapter submodule 2025-04-18 09:59:42 -06:00
losses.py Add stepped loss type 2025-09-22 15:50:12 -06:00
mask.py Added mask diffirential mask dialation for flex2. Handle video for the i2v adapter 2025-04-10 11:50:01 -06:00
nvfp4_quant.py Add support for MiniMax H3 T2V and I2V training 2026-08-03 10:17:39 -06:00
orbit_quant.py Fix casting issue with orbit quant 2026-07-10 10:23:38 -06:00
orbit_vq_quant.py Added experimental orbit quant 2026-07-09 15:14:51 -06:00
ostris_quant.py Add support for MiniMax H3 T2V and I2V training 2026-08-03 10:17:39 -06:00
quantize.py Speed up quantization processing on H3 2026-08-03 15:21:39 -06:00
qwen_vae_gradient_checkpointing.py Fix issue with qwen image edit models. 2026-07-17 09:14:04 -06:00
shuffle.py Various experiments and minor bug fixes for edge cases 2025-04-25 13:44:38 -06:00
uintx_quant.py Move uintx quantization to ostris quant with bit identical matching. Now we are not bound to an older version of torch ao. 2026-07-27 12:59:42 -06:00
vae.py Various bug fixes 2025-04-29 09:30:33 -06:00