src/transformers/models/speecht5/modeling_speecht5.py/SpeechT5PreTrainedModel/_set_gradient_checkpointing
class SpeechT5PreTrainedModel:    def _set_gradient_checkpointing(self, module, value=False):
        if isinstance(module, (SpeechT5Encoder, SpeechT5Decoder, SpeechT5FeatureEncoder)):
            module.gradient_checkpointing = value