From 83953a9a1870462b4097703804cbeb57b6180c57 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 14 Feb 2024 15:43:21 +0530 Subject: [PATCH] fix: controlnet inpaint single file. --- 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 233fa7dbb22d..5c5f34906094 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)