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

Commit
2 years ago
Add gradient checkpointing to Whisper Flax (#22954) * Add gradient checkpointing to Whisper Flax * self.gradient_checkpointing only needed in nn.Module, removing unnecessary comments
Author
Parents
Loading