It looks like you are using an older PyTorch version to load the model. Have a look at this thread for possible solutions.