src/transformers/generation_tf_utils.py/TFGenerationMixin/prepare_inputs_for_generation
class TFGenerationMixin:    def prepare_inputs_for_generation(self, inputs, **kwargs):
        """
        Implement in subclasses of :class:`~transformers.TFPreTrainedModel` for custom behavior to prepare inputs in
        the generate method.
        """
        return {"input_ids": inputs}