We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4343ce2 commit 37b0951Copy full SHA for 37b0951
src/diffusers/loaders/single_file_utils.py
@@ -1172,6 +1172,7 @@ def create_diffusers_unet_model_from_ldm(
1172
if num_in_channels is None:
1173
if pipeline_class_name in [
1174
"StableDiffusionInpaintPipeline",
1175
+ "StableDiffusionControlNetInpaintPipeline",
1176
"StableDiffusionXLInpaintPipeline",
1177
"StableDiffusionXLControlNetInpaintPipeline",
1178
]:
@@ -1190,6 +1191,7 @@ def create_diffusers_unet_model_from_ldm(
1190
1191
1192
diffusers_format_unet_checkpoint = convert_ldm_unet_checkpoint(checkpoint, unet_config, extract_ema=extract_ema)
1193
ctx = init_empty_weights if is_accelerate_available() else nullcontext
1194
+
1195
with ctx():
1196
unet = UNet2DConditionModel(**unet_config)
1197
0 commit comments