model_channels = state_dict['{}input_blocks.0.0.weight'.format(key_prefix)].shape[0]KeyError: 'model.diffusion_model.input_blocks.0.0.weight'
· Sign up or log in to comment
