src/transformers/models/pegasus/modeling_pegasus.py/PegasusForCausalLM/set_input_embeddings
class PegasusForCausalLM:    def set_input_embeddings(self, value):
        self.model.decoder.embed_tokens = value