code: def no_shard_zero_context(
      is_replicated: bool = True,
  ) -> AbstractContextManager:
      return ZeroContextMgr().hijack_context_config(
          target_device=torch.device(
              "cuda", torch.cuda.current_device()
          ),
          is_replicated=is_replicated,
          shard_param=False,
      )
  
imports: import contextlib
  import functools
  from contextlib import AbstractContextManager
  from dataclasses import dataclass
  from typing import Optional
  import torch
  import torch.distributed
  import torch.nn
  from colossalai.context.singleton_meta import SingletonMeta
  from colossalai.legacy.context.parallel_mode import ParallelMode
  from colossalai.legacy.core import global_context
  from colossalai.legacy.zero.shard_utils import BaseShardStrategy
  from colossalai.legacy.zero.sharded_model._utils import cast_tensor_to_bf16, cast_tensor_to_fp16
  from colossalai.legacy.zero.sharded_model.sharded_model_v2 import ShardedModelV2
  from colossalai.legacy.zero.sharded_param import ShardedParamV2
  from colossalai.logging import get_dist_logger
  from colossalai.utils.model.utils import InsertPostInitMethodToModuleSubClasses
label_desc: Use image node_id 1 for calling a global function with example usage: no_shard_zero_context(is_replicated) and returns: ZeroContextMgr