Jaret Burkett a4bbe167ce Added reference token attention isolation (kv_cache) for Krea2 edit training. Same training cost with significant inference speed up. 2x inference speedup.

2026-07-09 12:00:55 -06:00

..

__init__.py

Add support for Krea2 (#906)

2026-06-23 09:18:14 -06:00

mmdit.py

Added reference token attention isolation (kv_cache) for Krea2 edit training. Same training cost with significant inference speed up. 2x inference speedup.

2026-07-09 12:00:55 -06:00

pipeline.py

Added reference token attention isolation (kv_cache) for Krea2 edit training. Same training cost with significant inference speed up. 2x inference speedup.

2026-07-09 12:00:55 -06:00

text_encoder.py

Added support to train Krea2 as an edit model

2026-07-04 09:12:39 -06:00