src/transformers/models/gpt_neo/modeling_gpt_neo.py/GPTNeoForCausalLM/set_output_embeddings
class GPTNeoForCausalLM:    def set_output_embeddings(self, new_embeddings):
        self.lm_head = new_embeddings