src/transformers/models/xlm/modeling_tf_xlm.py/TFXLMWithLMHeadModel/__init__
class TFXLMWithLMHeadModel:    def __init__(self, config, *inputs, **kwargs):
        super().__init__(config, *inputs, **kwargs)
        self.transformer = TFXLMMainLayer(config, name="transformer")
        self.pred_layer = TFXLMPredLayer(config, self.transformer.embeddings, name="pred_layer_._proj")