src/transformers/models/ctrl/modeling_tf_ctrl.py/TFCTRLLMHead/__init__
class TFCTRLLMHead:    def __init__(self, config, input_embeddings, **kwargs):
        super().__init__(**kwargs)
        self.config = config
        # CTRL has numerical issues in XLA generate
        self.supports_xla_generation = False

        # The output weights are the same as the input embeddings, but there is
        # an output-only bias for each token.
        self.input_embeddings = input_embeddings