src/transformers/models/gpt_neox_japanese/modeling_gpt_neox_japanese.py/GPTNeoXJapaneseModel/set_input_embeddings
class GPTNeoXJapaneseModel:    def set_input_embeddings(self, value):
        self.embed_in = value