src/transformers/models/albert/modeling_tf_albert.py/TFAlbertForMaskedLM/__init__
class TFAlbertForMaskedLM:    def __init__(self, config: AlbertConfig, *inputs, **kwargs):
        super().__init__(config, *inputs, **kwargs)

        self.albert = TFAlbertMainLayer(config, add_pooling_layer=False, name="albert")
        self.predictions = TFAlbertMLMHead(config, input_embeddings=self.albert.embeddings, name="predictions")