diff --git a/src/diffusers/loaders.py b/src/diffusers/loaders.py index 2acfdc594ff8..1fbfd12f6db9 100644 --- a/src/diffusers/loaders.py +++ b/src/diffusers/loaders.py @@ -1123,7 +1123,7 @@ def _best_guess_weight_name(cls, pretrained_model_name_or_path_or_dict, file_ext @classmethod def _maybe_map_sgm_blocks_to_diffusers(cls, state_dict, unet_config, delimiter="_", block_slice_pos=5): # 1. get all state_dict_keys - all_keys = state_dict.keys() + all_keys = list(state_dict.keys()) sgm_patterns = ["input_blocks", "middle_block", "output_blocks"] # 2. check if needs remapping, if not return original dict