Hey! The model that you are loading is a key value dict with the key being the name of the layer and value the weights. You can add things to this dict, such as your new Linear layer. And yes, you can change the forward function to your liking 
There is some info on how to do the weights thingy in here