src/transformers/models/openai/modeling_tf_openai.py/TFOpenAIGPTLMHeadModel/prepare_inputs_for_generation
class TFOpenAIGPTLMHeadModel:    def prepare_inputs_for_generation(self, inputs, **kwargs):
        return {"input_ids": inputs}