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