code: def forward(self, x, arg_route=(True, False), **kwargs):
      f_args, g_args = map(
          lambda route: kwargs if route else {}, arg_route
      )
      block_kwargs = {"f_args": f_args, "g_args": g_args}
  
      return _ReversibleFunction.apply(x, self.blocks, block_kwargs)
  
imports: from typing import List
  import torch
  import torch.nn
  from torch.autograd.function import Function
  from torch.utils.checkpoint import get_device_states, set_device_states
  from xformers.components import RequiresWrappedInputs
label_desc: Use image node_id 2 for calling the ReversibleSequence obj's underlying member method code with example usage: obj.forward(x, arg_route) and returns: _ReversibleFunction