SIGN IN SIGN UP

feat(flux2klein): Add KV-cache acceleration for FLUX.2-Klein multi-image editing with E2E parity and smoke tests

- Implement prefix extraction phase (step 0) and cached denoising scan loop in FlaxFlux2KleinPipeline
- Support KV cache slicing and concatenation across Flax NNX Double and Single transformer blocks
- Safely bound Splash/Flash attention block sizes for asymmetric cross-attention sequences with opt-in flag
- Add CLI and config support for use_kv with dynamic FLUX.2-klein-9b-kv repository resolution
- Add cross-framework E2E parity test achieving 0.8881 SSIM / 22.58 dB PSNR
- Add 9B KV-cache image editing smoke test with verified golden reference image
A
amepas committed
8ad3cb5e2f91ce7c90f33e14e1103cb765f27d2a
Parent: e86be47