Wrapping a model into an nn.Sequential container might create some problems, if functional calls are using in the original forward method.
In your use case, you are losing the torch.flatten() operation from this line of code 113.
I would recommend to write a custom model class, derive from vgg as the base class, and manipulate the forward method as you wish.