src/transformers/models/speech_to_text/modeling_tf_speech_to_text.py/TFSpeech2TextForConditionalGeneration/__init__
class TFSpeech2TextForConditionalGeneration:    def __init__(self, config: Speech2TextConfig):
        super().__init__(config)
        self.model = TFSpeech2TextMainLayer(config, name="model")
        self.lm_head = tf.keras.layers.Dense(self.config.vocab_size, use_bias=False, name="lm_head")
        # TODO (Joao): investigate why Speech2Text has numerical issues in XLA generate
        self.supports_xla_generation = False