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