def __init__(self, config):
    super().__init__(config)
    self.transformer = GPTNeoModel(config)
    self.lm_head = LazyLinearAPICompatible(config.hidden_size, config.vocab_size, bias=False)
    self.init_weights()
def __init__(self, config):
    super().__init__(config)
    self.transformer = GPTNeoModel(config)
    self.lm_head = LazyLinearAPICompatible(config.hidden_size, config.vocab_size, bias=False)
    self.init_weights()