Adding Image Editing - #460

Closed

amepas wants to merge 5 commits into

mainfrom

onboarding-imageedit-flux2klein

Commits on Aug 14, 2026

feat(flux2klein): Onboard FLUX.2-Klein 4B and 9B with Flax NNX, fast weight loading, causal splash attention, and end-to-end optimizations
amepas
committed
3cb4455
Onboard FLUX.2-Klein multi-image editing support in pure Flax NNX (4B & 9B)
amepas
committed
284ff36

Commits on Aug 17, 2026

refactor(flux2klein): Unify text-to-image and multi-image editing into single pipeline and CLI
amepas
committed
1c4c1a9
cleanup(flux2klein): Remove redundant standalone image-edit files and obsolete dev tests
amepas
committed
c20a7dc
test(flux2klein): Add 4B/9B image edit smoke tests and rename E2E test
amepas
committed
622fc63