code: def get_states(copy=False):
      """Returns the seed states of the seed manager.
  
      Returns:
          dict: The seed states of the seed manager.
      """
      states = _SEED_MANAGER.seed_states
  
      if copy:
          new_states = dict()
  
          for parallel_mode, state in states.items():
              new_states[parallel_mode] = state.clone()
          return new_states
      else:
          return _SEED_MANAGER.seed_states
  
file_path: ColossalAI/colossalai/legacy/context/random/_helper.py
imports: import functools
  from contextlib import contextmanager
  import torch.cuda
  from torch import Tensor
  from ..parallel_mode import ParallelMode
  from .seed_manager import SeedManager
label_desc: Use image node_id 2 for calling a global function with example usage: get_states(copy) and returns: new_states, _SEED_MANAGER