src/transformers/models/rembert/modeling_rembert.py/RemBertPreTrainedModel/_set_gradient_checkpointing
class RemBertPreTrainedModel:    def _set_gradient_checkpointing(self, module, value=False):
        if isinstance(module, RemBertEncoder):
            module.gradient_checkpointing = value