src/transformers/models/convnextv2/modeling_convnextv2.py/ConvNextV2DropPath/forward
class ConvNextV2DropPath:    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
        return drop_path(hidden_states, self.drop_prob, self.training)