transformers

ba0dc545 - Add gradient checkpointing to Whisper Flax (#22954)

Add gradient checkpointing to Whisper Flax (#22954) * Add gradient checkpointing to Whisper Flax * self.gradient_checkpointing only needed in nn.Module, removing unnecessary comments