src/transformers/models/blenderbot_small/modeling_tf_blenderbot_small.py/TFBlenderbotSmallMainLayer/set_input_embeddings
class TFBlenderbotSmallMainLayer:    def set_input_embeddings(self, new_embeddings):
        self.shared = new_embeddings
        self.encoder.embed_tokens = self.shared
        self.decoder.embed_tokens = self.shared