AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

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

Browse Source 
* Update pipeline_controlnet_inpaint.py

* Apply style fixes
This commit is contained in:
CyberVy
2025-03-21 10:56:12 +08:00
committed by GitHub
parent f424b1b062
commit 9b2c0a7dbe
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py
View File

@@ -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,