def set_output_embeddings(self, value):
    self.input_embeddings.weight = value
    self.input_embeddings.vocab_size = shape_list(value)[0]
def set_output_embeddings(self, value):
    self.input_embeddings.weight = value
    self.input_embeddings.vocab_size = shape_list(value)[0]