def gradient_checkpointing_enable()

in weak_to_strong/model.py [0:0]

• 5 lines of code
• 2 McCabe index (conditional complexity)

    def gradient_checkpointing_enable(self):
        model = self.transformer
        (
            model if hasattr(model, "save_pretrained") else model.module
        ).gradient_checkpointing_enable()