src/transformers/models/longt5/modeling_flax_longt5.py/FlaxLongT5ForConditionalGeneration/update_inputs_for_generation
class FlaxLongT5ForConditionalGeneration:    def update_inputs_for_generation(self, model_outputs, model_kwargs):
        model_kwargs["past_key_values"] = model_outputs.past_key_values
        return model_kwargs