code: def _format_memory(nbytes) -> str:
      """Returns a formatted memory size string"""
      KB = 1024
      MB = 1024 * KB
      GB = 1024 * MB
      if abs(nbytes) >= GB:
          return "{:.2f} Gb".format(nbytes * 1.0 / GB)
      elif abs(nbytes) >= MB:
          return "{:.2f} Mb".format(nbytes * 1.0 / MB)
      elif abs(nbytes) >= KB:
          return "{:.2f} Kb".format(nbytes * 1.0 / KB)
      else:
          return str(nbytes) + " b"
  
imports: from typing import Any, Callable, Dict, List, Tuple, Optional, Union
  from dataclasses import dataclass, field
  import torch
  import torch.fx
  from torch.fx import Interpreter
  from torch.fx.node import Argument, Node
  from .flop_utils import flop_count
label_desc: Use image node_id 2 for calling a global function with example usage: _format_memory(nbytes) and returns: str, str, str, unknown