From 303d5641baa99478c91acdb55f1f3f95bdebe0ee Mon Sep 17 00:00:00 2001 From: Aleri Kaisattera <73682764+alerikaisattera@users.noreply.github.com> Date: Wed, 3 Sep 2025 19:30:57 +0600 Subject: [PATCH] Flex.2: fix pipeline bug that forces contol in inpaint Flex.2 pipeline has a bug that forces use of a control image alongside inpaint image This fix comes from HF user katuni4ka Link: https://huggingface.co/ostris/Flex.2-preview/discussions/16 --- pipelines/flex2/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pipelines/flex2/__init__.py b/pipelines/flex2/__init__.py index 6c452551f..dbb8e2a0d 100644 --- a/pipelines/flex2/__init__.py +++ b/pipelines/flex2/__init__.py @@ -291,7 +291,7 @@ class Flex2Pipeline(FluxControlPipeline): ) inpaint_image = self.vae.encode(inpaint_image).latent_dist.sample(generator=generator) inpaint_latents = (inpaint_image - self.vae.config.shift_factor) * self.vae.config.scaling_factor - height_inpaint_image, width_inpaint_image = control_image.shape[2:] + height_inpaint_image, width_inpaint_image = inpaint_image.shape[2:] inpaint_mask = self.prepare_image( image=inpaint_mask,