code: def get_dtypes(include_float=True):
      global DTYPES
      if DTYPES is None:
          DTYPES = (
              [torch.float16, torch.float32]
              if include_float
              else [torch.float16]
          )
          try:
              if get_accelerator().is_bf16_supported():
                  DTYPES.append(torch.bfloat16)
          except (AssertionError, AttributeError):
              pass
      return DTYPES
  
imports: from typing import Tuple
  import torch
  from deepspeed.accelerator import get_accelerator
label_desc: Use image node_id 2 for calling a global function with example usage: get_dtypes(include_float) and returns: DTYPES