(module)
| 416 | if getattr(self.config, "gradient_checkpointing", False): |
| 417 | |
| 418 | def create_custom_forward(module): |
| 419 | def custom_forward(*inputs): |
| 420 | return module(*inputs, output_attentions) |
| 421 | |
| 422 | return custom_forward |
| 423 | |
| 424 | layer_outputs = torch.utils.checkpoint.checkpoint( |
| 425 | create_custom_forward(layer_module), |
nothing calls this directly
no outgoing calls
no test coverage detected