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
This commit is contained in:
Aleri Kaisattera
2025-09-03 19:30:57 +06:00
committed by GitHub
parent 3ec0d35d89
commit 303d5641ba
+1 -1
View File
@@ -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,