src/transformers/models/t5/modeling_t5.py/T5PreTrainedModel/_set_gradient_checkpointing
class T5PreTrainedModel:    def _set_gradient_checkpointing(self, module, value=False):
        if isinstance(module, (T5Attention, T5Stack)):
            module.gradient_checkpointing = value