You need to sign in or sign up before continuing.
Unverified Commit c1b2a89e authored by yunseong Cho's avatar yunseong Cho Committed by GitHub
Browse files

Fix key error for dictionary with randomized order in convert_ldm_unet_checkpoint (#7680)



fix key error for different order
Co-authored-by: default avataryunseong <yunseong.cho@superlabs.us>
Co-authored-by: default avatarDhruv Nair <dhruv.nair@gmail.com>
parent 435d37ce
...@@ -557,7 +557,7 @@ def convert_ldm_unet_checkpoint( ...@@ -557,7 +557,7 @@ def convert_ldm_unet_checkpoint(
paths, new_checkpoint, unet_state_dict, additional_replacements=[meta_path], config=config 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(): if ["conv.bias", "conv.weight"] in output_block_list.values():
index = list(output_block_list.values()).index(["conv.bias", "conv.weight"]) 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[ new_checkpoint[f"up_blocks.{block_id}.upsamplers.0.conv.weight"] = unet_state_dict[
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment