src/transformers/models/gpt_neo/modeling_gpt_neo.py/GPTNeoPreTrainedModel/_set_gradient_checkpointing
class GPTNeoPreTrainedModel:    def _set_gradient_checkpointing(self, module, value=False):
        if isinstance(module, GPTNeoModel):
            module.gradient_checkpointing = value