The mentioned issue is not caused by different PyTorch versions, but would be a wrong usage of the .module attribute.
Sure, please describe the issue in more detail and, if possible, post a minimal, executable code snippet so that we could debug it.