Fix the model of transformer