code: def format_num(num: int, bytes=False):
      """Scale bytes to its proper format, e.g. 1253656 => '1.20MB'"""
      factor = 1024 if bytes else 1000
      suffix = "B" if bytes else ""
      for unit in ["", " K", " M", " G", " T", " P"]:
          if num < factor:
              return f"{num:.2f}{unit}{suffix}"
          num /= factor
  
file_path: ColossalAI/examples/language/opt/opt_benchmark.py
imports: import time
  import torch
  import tqdm
  import transformers
  from args import parse_benchmark_args
  from transformers import AutoConfig, OPTForCausalLM
  from transformers.utils.versions import require_version
  import colossalai
  from colossalai.booster import Booster
  from colossalai.booster.plugin import GeminiPlugin, LowLevelZeroPlugin, TorchDDPPlugin
  from colossalai.cluster import DistCoordinator
  from colossalai.logging import disable_existing_loggers, get_dist_logger
  from colossalai.nn.optimizer import HybridAdam
label_desc: Use image node_id 1 for calling a global function with example usage: format_num(num, bytes) and returns: num, unit, suffix