Adding Image Editing to Flux2.Klein Models - #461

Open

amepas wants to merge 6 commits into

mainfrom

onboarding-imageedit-flux2klein

Commits on Aug 22, 2026

feat(flux2klein): Onboard multi-image editing support in pure Flax NNX (4B & 9B) with E2E parity and smoke tests
amepas
committed
34d1b0f

Commits on Aug 24, 2026

fix(transformers): make image edit branch fully compatible with transformers 5.x without accuracy regression
amepas
committed
856fb45

Commits on Aug 25, 2026

fix(flux2klein): resolve multi-image edit pipeline, add preprocessing unit tests, and fix ruff/pyink compliance
amepas
committed
35d8c77
refactor(tests): resolve dynamic HF_HOME model snapshots in generate_flux2klein_e2e_test.py
amepas
committed
9e84d35
refactor(tests): strictly use HF_HOME without hardcoded environment paths
amepas
committed
84dc112

Commits on Aug 26, 2026

fix(compat): isolate Flux2.Klein import closure for Transformers 4.x & 5.x and revert broad legacy edits
amepas
committed
9136788