src/transformers/models/wav2vec2/modeling_wav2vec2.py/Wav2Vec2PreTrainedModel/_set_gradient_checkpointing
class Wav2Vec2PreTrainedModel:    def _set_gradient_checkpointing(self, module, value=False):
        if isinstance(module, (Wav2Vec2Encoder, Wav2Vec2EncoderStableLayerNorm, Wav2Vec2FeatureEncoder)):
            module.gradient_checkpointing = value