def __init__(self, config, input_embeddings, **kwargs):
    super().__init__(**kwargs)
    self.vocab_size = config.vocab_size
    self.embedding_size = config.embedding_size
    self.input_embeddings = input_embeddings
def __init__(self, config, input_embeddings, **kwargs):
    super().__init__(**kwargs)
    self.vocab_size = config.vocab_size
    self.embedding_size = config.embedding_size
    self.input_embeddings = input_embeddings