Skip to content

Commit 9b2c0a7

Browse files
authored
fix _callback_tensor_inputs of sd controlnet inpaint pipeline missing some elements (#11073)
* Update pipeline_controlnet_inpaint.py * Apply style fixes
1 parent f424b1b commit 9b2c0a7

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,14 @@ class StableDiffusionControlNetInpaintPipeline(
184184
model_cpu_offload_seq = "text_encoder->image_encoder->unet->vae"
185185
_optional_components = ["safety_checker", "feature_extractor", "image_encoder"]
186186
_exclude_from_cpu_offload = ["safety_checker"]
187-
_callback_tensor_inputs = ["latents", "prompt_embeds", "negative_prompt_embeds", "control_image"]
187+
_callback_tensor_inputs = [
188+
"latents",
189+
"prompt_embeds",
190+
"negative_prompt_embeds",
191+
"control_image",
192+
"mask",
193+
"masked_image_latents",
194+
]
188195

189196
def __init__(
190197
self,

0 commit comments

Comments
 (0)