src/transformers/models/gptj/modeling_gptj.py/GPTJPreTrainedModel/_set_gradient_checkpointing
class GPTJPreTrainedModel:    def _set_gradient_checkpointing(self, module, value=False):
        if isinstance(module, GPTJModel):
            module.gradient_checkpointing = value