AUTOMATIC1111/stable-diffusion-webui-pixelization

Error(s) in loading state_dict

WorldofDepth opened this issue · 3 comments

WorldofDepth commented

Any idea why I get this error? I've downloaded the three checkpoint models, and should be fine on RAM… Thank you for any help.

RuntimeError: Error(s) in loading state_dict for DataParallel: Missing key(s) in state_dict: "module.PBEnc.vgg.0.weight", "module.PBEnc.vgg.0.bias" [continues to list many files]

Azq2 commented

image

This should fix

            self.G_A_net = torch.nn.DataParallel(self.G_A_net)
            self.alias_net = torch.nn.DataParallel(self.alias_net)
WorldofDepth commented

Thank you. Where / which file should that be inserted into?