src/transformers/models/mbart/modeling_tf_mbart.py/TFMBartMainLayer/set_input_embeddings
class TFMBartMainLayer:    def set_input_embeddings(self, new_embeddings):
        self.shared = new_embeddings
        self.encoder.embed_tokens = self.shared
        self.decoder.embed_tokens = self.shared