Add FLUX.2 Klein Inpaint Pipeline (#13050)
* Add Flux2KleinInpaintPipeline
* Fixed mask channel mismatch and a bit of cleaning
* Added tests and minor refactors
* Added support for reference images for inpainting
* Style fixes
* Fixed the example docstring
* Corrected mask latent preparation for correct dimensional alignment
* replace masked_image_latents context with clean_source_latents, fix mask spatial alignment and remove unused VAE encoding
* Fix T-coordinate collision for conditioning
* Changed the default strength from 0.6 to 0.8
* Added reference image test and updated the frozenset
* Validated ref image, latent passing support and fixed ref image preprocessing
* Refined preprocessing with 1MP resolution cap and timestep tracking
* Updated typing, improved validation and changed the example docstring
* Style fixes
* Fixed batch inference discrepancy and addressed review comments
* Fixed a typo
Co-authored-by: Álvaro Somoza <asomoza@users.noreply.github.com>
* Apply suggestion from @asomoza
Co-authored-by: Álvaro Somoza <asomoza@users.noreply.github.com>
* Reused encoded latents and fix channel check consistency
* fixed pre-encoded latent preprocessing for source and ref images
* Apply style fixes
* Updated the docstring with the shape requirements
* Apply style fixes
* Fixed copies