Using Trainer to save a Bartforsequenceclassification model

It looks like you have introduced a new layer in your model. Is it so?