Problems with the Weight of resolution 256 stage2 #24
Description
The released stage 2 weight of resolution 256 seems to be incomplete, the error log is shown below.
File "/home/user/data/PT/PCDMs/stage2_batchtest_inpaint_model.py", line 126, in inference pose_proj.load_state_dict(pose_proj_dict)
RuntimeError: Error(s) in loading state_dict for ControlNetConditioningEmbedding: Missing key(s) in state_dict: "conv_in.weight", "conv_in.bias", "blocks.0.weight", "blocks.0.bias", "blocks.1.weight", "blocks.1.bias", "blocks.2.weight", "blocks.2.bias", "blocks.3.weight", "blocks.3.bias", "blocks.4.weight", "blocks.4.bias", "blocks.5.weight", "blocks.5.bias", "conv_out.weight", "conv_out.bias".
I also find the stage 2 .pt file is quite different between resolution 256 and 512.
Maybe its actually the stage 1 weight of 256?
Could you please help check this file or I wrongly process this step? thank you very much.