src/transformers/models/nystromformer/modeling_nystromformer.py/NystromformerPreTrainedModel/_set_gradient_checkpointing
class NystromformerPreTrainedModel:    def _set_gradient_checkpointing(self, module, value=False):
        if isinstance(module, NystromformerEncoder):
            module.gradient_checkpointing = value