Using the mentioned approach by re-wrapping modules into an nn.Sequential container might break for models which are using the functional API in their forward or any other approach which is not initializing and applying all layers in a sequential way.
In your case, you are missing these functional API calls 3, which is why the output shape differs.