code: def _set_checkpoint_device_map(device_map):
      global CHECKPOINT_DEVICE_MAP
      CHECKPOINT_DEVICE_MAP = device_map
  
imports: from typing import Tuple, Union
  import torch
  import dill
label_desc: Use image node_id 1 for calling a global function with example usage: _set_checkpoint_device_map(device_map) without return types