Error(s) in loading state_dict

[WorldofDepth]

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]

This should fix

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

[WorldofDepth]

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

[Azq2]

https://github.com/AUTOMATIC1111/stable-diffusion-webui-pixelization/blob/master/scripts/postprocessing_pixelization.py#L126