(module)
| 698 | if self.gradient_checkpointing and self.training: |
| 699 | |
| 700 | def create_custom_forward(module): |
| 701 | def custom_forward(*inputs): |
| 702 | # None for past_key_value |
| 703 | return module(*inputs, use_cache=use_cache, output_attentions=output_attentions) |
| 704 | |
| 705 | return custom_forward |
| 706 | |
| 707 | outputs = torch.utils.checkpoint.checkpoint( |
| 708 | create_custom_forward(block), |
nothing calls this directly
no outgoing calls
no test coverage detected