Based on the error message it seems that the created model via pg.Generator(depth=6) does not have the same parameters as the stored model.
I’m not familiar with the repository, so cannot comment on the model architectures.
It seems that module.rgb_converters might be nn.Conv2d layers, which raise the shape mismatch error in the output filter dimension?
Also, the state_dict seem to contain additional layers (module.layers.4). Could you have trained the model “longer”, such that more layers were added during the progressive training?