[Bug] Controlnet weights are saved as 'controlnet.module.model...' but are indeed loaded as 'controlnet.model...'. about mmagic HOT 2 CLOSED

Sent from my iPhone On Jul 31, 2023, at 11:22, LeoXing1996 ***@***.***> wrote:  @LoveU3tHousand2<https://github.com/LoveU3tHousand2>, sorry for late response. You can process the checkpoint file with the following script: import torch new_model_dict = dict() state_dict = torch.load(CHECKPOINT) model_dict = state_dict['model'] for k, v in model_dict.items(): if k.startswith('controlnet'): new_k = k.replace('module.', '') new_model_dict[new_k] = v else: new_model_dict[k] = v state_dict['model'] = new_model_dict — Reply to this email directly, view it on GitHub<#1959 (comment)>, or unsubscribe<https://github.com/notifications/unsubscribe-auth/AQTQZ5EWXV5I23DUGEQGA23XS4QIFANCNFSM6AAAAAA2ZMA47A>. You are receiving this because you were mentioned.Message ID: ***@***.***>