From 37b09517b91a2efd62a60ee676f9ebeddfd86c30 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Wed, 14 Feb 2024 19:04:57 +0530 Subject: [PATCH] fix: controlnet inpaint single file. (#6975) --- src/diffusers/loaders/single_file_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/diffusers/loaders/single_file_utils.py b/src/diffusers/loaders/single_file_utils.py index 233fa7dbb2..5c5f349060 100644 --- a/src/diffusers/loaders/single_file_utils.py +++ b/src/diffusers/loaders/single_file_utils.py @@ -1172,6 +1172,7 @@ def create_diffusers_unet_model_from_ldm( if num_in_channels is None: if pipeline_class_name in [ "StableDiffusionInpaintPipeline", + "StableDiffusionControlNetInpaintPipeline", "StableDiffusionXLInpaintPipeline", "StableDiffusionXLControlNetInpaintPipeline", ]: @@ -1190,6 +1191,7 @@ def create_diffusers_unet_model_from_ldm( diffusers_format_unet_checkpoint = convert_ldm_unet_checkpoint(checkpoint, unet_config, extract_ema=extract_ema) ctx = init_empty_weights if is_accelerate_available() else nullcontext + with ctx(): unet = UNet2DConditionModel(**unet_config)