(module)
| 507 | use_cache = False |
| 508 | |
| 509 | def create_custom_forward(module): |
| 510 | def custom_forward(*inputs): |
| 511 | return module(*inputs, past_key_value, output_attentions) |
| 512 | |
| 513 | return custom_forward |
| 514 | |
| 515 | layer_outputs = torch.utils.checkpoint.checkpoint( |
| 516 | create_custom_forward(layer_module), |
nothing calls this directly
no outgoing calls
no test coverage detected