diffusers
74482585 - Add FLUX.2 Klein Inpaint Pipeline (#13050)

Commit
86 days ago
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
Author
Parents
Loading