(module)
| 839 | |
| 840 | if self.training and self.gradient_checkpointing: |
| 841 | def create_custom_forward(module): |
| 842 | def custom_forward(*inputs): |
| 843 | return module(*inputs) |
| 844 | |
| 845 | return custom_forward |
| 846 | |
| 847 | hidden_states = torch.utils.checkpoint.checkpoint(create_custom_forward(resnet), hidden_states, temb) |
| 848 | if motion_module is not None: |
nothing calls this directly
no outgoing calls
no test coverage detected