Skip to content

Commit 2971b18

Browse files
authored
Update pipeline_controlnet_inpaint.py
1 parent 1001425 commit 2971b18

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ 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 = ["latents", "prompt_embeds", "negative_prompt_embeds", "control_image", "mask", "masked_image_latents"]
188188

189189
def __init__(
190190
self,

0 commit comments

Comments
 (0)