* (mask_condition < 0.5)

hlky · hlky · commit efadd91f8b49 · 2025-12-13T13:27:47.000Z
diff --git a/src/diffusers/pipelines/z_image/pipeline_z_image_controlnet_inpaint.py b/src/diffusers/pipelines/z_image/pipeline_z_image_controlnet_inpaint.py
@@ -566,6 +566,7 @@ def __call__(
             dtype=self.vae.dtype,
         )
         height, width = init_image.shape[-2:]
+        init_image = init_image * (mask_condition < 0.5)
         init_image = retrieve_latents(self.vae.encode(init_image), generator=generator, sample_mode="argmax")
         init_image = (init_image - self.vae.config.shift_factor) * self.vae.config.scaling_factor
         init_image = init_image.unsqueeze(2)

Original file line number	Diff line number	Diff line change
`@@ -566,6 +566,7 @@ def __call__(`
`566`	`566`	`dtype=self.vae.dtype,`
`567`	`567`	`)`
`568`	`568`	`height, width = init_image.shape[-2:]`
	`569`	`+ init_image = init_image * (mask_condition < 0.5)`
`569`	`570`	`init_image = retrieve_latents(self.vae.encode(init_image), generator=generator, sample_mode="argmax")`
`570`	`571`	`init_image = (init_image - self.vae.config.shift_factor) * self.vae.config.scaling_factor`
`571`	`572`	`init_image = init_image.unsqueeze(2)`