Skip to content

Commit 37b0951

Browse files
authored
fix: controlnet inpaint single file. (#6975)
1 parent 4343ce2 commit 37b0951

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/diffusers/loaders/single_file_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1172,6 +1172,7 @@ def create_diffusers_unet_model_from_ldm(
11721172
if num_in_channels is None:
11731173
if pipeline_class_name in [
11741174
"StableDiffusionInpaintPipeline",
1175+
"StableDiffusionControlNetInpaintPipeline",
11751176
"StableDiffusionXLInpaintPipeline",
11761177
"StableDiffusionXLControlNetInpaintPipeline",
11771178
]:
@@ -1190,6 +1191,7 @@ def create_diffusers_unet_model_from_ldm(
11901191

11911192
diffusers_format_unet_checkpoint = convert_ldm_unet_checkpoint(checkpoint, unet_config, extract_ema=extract_ema)
11921193
ctx = init_empty_weights if is_accelerate_available() else nullcontext
1194+
11931195
with ctx():
11941196
unet = UNet2DConditionModel(**unet_config)
11951197

0 commit comments

Comments
 (0)