def __init__(self, config, *inputs, **kwargs):
    super().__init__(config, **kwargs)
    self.vocab_size = config.vocab_size
    self.convbert = TFConvBertMainLayer(config, name='convbert')
    self.generator_predictions = TFConvBertGeneratorPredictions(config, name='generator_predictions')
    if isinstance(config.hidden_act, str):
        self.activation = get_tf_activation(config.hidden_act)
    else:
        self.activation = config.hidden_act
    self.generator_lm_head = TFConvBertMaskedLMHead(config, self.convbert.embeddings, name='generator_lm_head')
def __init__(self, config, *inputs, **kwargs):
    super().__init__(config, **kwargs)
    self.vocab_size = config.vocab_size
    self.convbert = TFConvBertMainLayer(config, name='convbert')
    self.generator_predictions = TFConvBertGeneratorPredictions(config, name='generator_predictions')
    if isinstance(config.hidden_act, str):
        self.activation = get_tf_activation(config.hidden_act)
    else:
        self.activation = config.hidden_act
    self.generator_lm_head = TFConvBertMaskedLMHead(config, self.convbert.embeddings, name='generator_lm_head')