From 9b2c0a7dbe5487e700a0039a09c277d73a17ccc2 Mon Sep 17 00:00:00 2001
From: CyberVy <72680847+CyberVy@users.noreply.github.com>
Date: Fri, 21 Mar 2025 10:56:12 +0800
Subject: [PATCH] fix _callback_tensor_inputs of sd controlnet inpaint pipeline
 missing some elements (#11073)

* Update pipeline_controlnet_inpaint.py

* Apply style fixes
---
 .../pipelines/controlnet/pipeline_controlnet_inpaint.py  | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py
index 40092e5f47..16d3529ed3 100644
--- 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,