src/transformers/models/vit_msn/modeling_vit_msn.py/ViTMSNPreTrainedModel/_set_gradient_checkpointing
class ViTMSNPreTrainedModel:    def _set_gradient_checkpointing(self, module: ViTMSNEncoder, value: bool = False) -> None:
        if isinstance(module, ViTMSNEncoder):
            module.gradient_checkpointing = value