code: def _get_autocast_dtype(precision: str):
      if precision in ["bf16", "bf16-mixed"]:
          return torch.bfloat16
      if precision in [32, "32", "32-true"]:
          return torch.float
      if precision in [16, "16", "16-mixed"]:
          return torch.half
      raise ValueError(
          'precision must be in ["32-true", "16-mixed", "bf16-mixed"]'
      )
  
file_path: NeMo/examples/multimodal/vision_language_foundation/nsfw/megatron_nsfw_infer.py
imports: import torch
  from omegaconf.omegaconf import OmegaConf
  from PIL import Image
  from nemo.collections.multimodal.data.clip.augmentations.augmentations import image_transform
  from nemo.collections.multimodal.models.vision_language_foundation.megatron_nsfw_clip_models import MegatronContentFilteringModel
  from nemo.collections.multimodal.parts.utils import setup_trainer_and_model_for_inference
  from nemo.core.config import hydra_runner
  from nemo.utils import logging
  from nemo.utils.get_rank import is_global_rank_zero
label_desc: Use image node_id 1 for calling a global function with example usage: _get_autocast_dtype(precision) and returns: torch, torch, torch