code: def set_seed_states(parallel_mode: ParallelMode, state: Tensor):
      """Sets the state of the seed manager for `parallel_mode`.
  
      Args:
          parallel_mode (:class:`colossalai.legacy.context.ParallelMode`): The chosen parallel mode.
          state (:class:`torch.Tensor`): the state to be set.
  
      Raises:
          AssertionError: Raises an AssertionError if `parallel_mode` is not found in the seed manager.
      """
      _SEED_MANAGER.set_state(parallel_mode, state)
  
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 6 for calling a global function with example usage: set_seed_states(parallel_mode, state) without return types