From c1b2a89e349b3b632c1f32052331e549516f87fc Mon Sep 17 00:00:00 2001 From: yunseong Cho Date: Thu, 2 May 2024 13:59:55 +0900 Subject: [PATCH] Fix key error for dictionary with randomized order in convert_ldm_unet_checkpoint (#7680) fix key error for different order Co-authored-by: yunseong Co-authored-by: Dhruv Nair --- src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py index f04a21ef48..b4b1b885dd 100644 --- a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +++ b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py @@ -557,7 +557,7 @@ def convert_ldm_unet_checkpoint( paths, new_checkpoint, unet_state_dict, additional_replacements=[meta_path], config=config ) - output_block_list = {k: sorted(v) for k, v in output_block_list.items()} + output_block_list = {k: sorted(v) for k, v in sorted(output_block_list.items())} if ["conv.bias", "conv.weight"] in output_block_list.values(): index = list(output_block_list.values()).index(["conv.bias", "conv.weight"]) new_checkpoint[f"up_blocks.{block_id}.upsamplers.0.conv.weight"] = unet_state_dict[