Mismatch between the model and the chekcpoint
#3
by
suko87tr
- opened
RuntimeError: Error(s) in loading state_dict for ControlLDM:
size mismatch for control_model.input_hint_block.0.weight: copying a param with shape torch.Size([16, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([16, 5, 3, 3]).
The provided colorizenet-sd21.ckpt is in [16, 3, 3, 3] shape whereas the model is expecting [16, 5, 3, 3].
I have already modified the in_hint_ch = 3 in yaml file but if I do so, it doesn't process with colorizing.
Thanks in advance if you have a solution.