I am working on a code where I have received a pre-trained model from one of my team members. The model is trained on PyTorch and does translation tasks. Is there a way that