src/transformers/models/bloom/modeling_bloom.py/BloomPreTrainedModel/_set_gradient_checkpointing
class BloomPreTrainedModel:    def _set_gradient_checkpointing(self, module: nn.Module, value: bool = False):
        if isinstance(module, BloomModel):
            module.gradient_checkpointing = value