code: def _unlift_exported_program_lifted_states(
      ep: ExportedProgram,
  ) -> torch.nn.Module:
      new_gm = copy.deepcopy(ep.graph_module)
      inp_pos_to_param_buffer_name = (
          _construct_inp_pos_to_param_buffer_name(
              new_gm,
              ep.graph_signature,
              ep.state_dict,
              ep.tensor_constants,
          )
      )
      new_gm = _unlift(
          new_gm,
          inp_pos_to_param_buffer_name,
          ep.call_spec.in_spec,
          ep.call_spec.out_spec,
          ep.state_dict,
          ep.tensor_constants,
          ep.graph_signature.buffers_to_mutate,
      )
      unlift_gm = _create_stateful_graph_module(
          new_gm, ep.range_constraints
      )
      unlift_gm.meta.update(ep.graph_module.meta)
      return unlift_gm
  
file_path: pytorch/torch/export/_unlift.py
imports: import copy
  import torch
  import torch.utils._pytree
  from torch._export.utils import _check_input_constraints_pre_hook
  from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo
  from .exported_program import ExportedProgram
label_desc: Use image node_id 4 for calling a global function with example usage: _unlift_exported_program_lifted_states(ep) and returns: unlift_gm