src/transformers/models/t5/modeling_flax_t5.py/FlaxT5ForConditionalGeneration/update_inputs_for_generation
class FlaxT5ForConditionalGeneration:    def update_inputs_for_generation(self, model_outputs, model_kwargs):
        model_kwargs["past_key_values"] = model_outputs.past_key_values
        return model_kwargs