Jaret Burkett f1bc6508ad Fix issue with qwen image edit models.		2026-07-17 09:14:04 -06:00
..
blended_blur_noise.py	Various experiments and minor bug fixes for edge cases	2025-04-25 13:44:38 -06:00
convrot_quant.py	Made a fused GEMV kernel for convrot unpacking to increase speed further. Fix bug in test script that made train time add additional grads to bf16.	2026-07-15 10:24:52 -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
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	Do not materalize weights on ostris quantizer when getting state dict. Require dequantization of weight like other quantization methods.	2026-07-16 08:55:51 -06:00
quantize.py	Do not materalize weights on ostris quantizer when getting state dict. Require dequantization of weight like other quantization methods.	2026-07-16 08:55:51 -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
vae.py	Various bug fixes	2025-04-29 09:30:33 -06:00