fix _callback_tensor_inputs of sd controlnet inpaint pipeline missing some elements (#11073)

* Update pipeline_controlnet_inpaint.py * Apply style fixes

fix _callback_tensor_inputs of sd controlnet inpaint pipeline missing some elements (#11073)
* Update pipeline_controlnet_inpaint.py * Apply style fixes
9b2c0a7d · CyberVy · GitHub · f424b1b0 · 9b2c0a7d
Unverified Commit 9b2c0a7d authored Mar 21, 2025 by CyberVy Committed by GitHub Mar 20, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 1 deletion

src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py ...users/pipelines/controlnet/pipeline_controlnet_inpaint.py +8 -1

No files found.
--- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py
+++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py
@@ -184,7 +184,14 @@ class StableDiffusionControlNetInpaintPipeline(
    model_cpu_offload_seq = "text_encoder->image_encoder->unet->vae"
    _optional_components = ["safety_checker", "feature_extractor", "image_encoder"]
    _exclude_from_cpu_offload = ["safety_checker"]
-    _callback_tensor_inputs = ["latents", "prompt_embeds", "negative_prompt_embeds", "control_image"]
+    _callback_tensor_inputs = [
+        "latents",
+        "prompt_embeds",
+        "negative_prompt_embeds",
+        "control_image",
+        "mask",
+        "masked_image_latents",
+    ]

    def __init__(
        self,