code: def set_checkpoint_version(value):
      global _CHECKPOINT_VERSION
      if _CHECKPOINT_VERSION is not None:
          assert (
              _CHECKPOINT_VERSION == value
          ), "checkpoint versions do not match"
      _CHECKPOINT_VERSION = value
  
imports: import os
  import random
  import sys
  import numpy
  import torch
  from megatron import update_num_microbatches
  from megatron.core import mpu, tensor_parallel
  from .global_vars import get_args
  from .utils import unwrap_model, print_rank_0
label_desc: Use image node_id 1 for calling a global function with example usage: set_checkpoint_version(value) without return types