From 19df9f3ec0ff9d3f1b6fb62cfb4867294f7bca2a Mon Sep 17 00:00:00 2001 From: Nikita Date: Mon, 20 May 2024 19:24:49 +0300 Subject: [PATCH] Update pipeline_controlnet_inpaint_sd_xl.py (#7983) --- .../pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py index 56c2dc10a5..966196b446 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py @@ -227,7 +227,7 @@ class StableDiffusionXLControlNetInpaintPipeline( tokenizer: CLIPTokenizer, tokenizer_2: CLIPTokenizer, unet: UNet2DConditionModel, - controlnet: ControlNetModel, + controlnet: Union[ControlNetModel, List[ControlNetModel], Tuple[ControlNetModel], MultiControlNetModel], scheduler: KarrasDiffusionSchedulers, requires_aesthetics_score: bool = False, force_zeros_for_empty_prompt: bool = True,