src/transformers/models/rembert/modeling_tf_rembert.py/TFRemBertForCausalLM/__init__
class TFRemBertForCausalLM:    def __init__(self, config: RemBertConfig, *inputs, **kwargs):
        super().__init__(config, *inputs, **kwargs)

        if not config.is_decoder:
            logger.warning("If you want to use `TFRemBertForCausalLM` as a standalone, add `is_decoder=True.`")

        self.rembert = TFRemBertMainLayer(config, name="rembert", add_pooling_layer=False)
        self.mlm = TFRemBertMLMHead(config, input_embeddings=self.rembert.embeddings, name="mlm___cls")