src/transformers/models/gpt_neox_japanese/modeling_gpt_neox_japanese.py/GPTNeoXJapaneseForCausalLM/set_output_embeddings
class GPTNeoXJapaneseForCausalLM:    def set_output_embeddings(self, new_embeddings):
        self.embed_out = new_embeddings