code: def _get_world_size() -> int:
      return int(os.environ.get("LOCAL_WORLD_SIZE", "1"))
  
file_path: gpt-fast/tp.py
imports: import os
  from typing import List, Optional
  import torch
  import torch.distributed
  from torch import nn
  from torch.distributed import _functional_collectives
  from model import Attention, FeedForward, Transformer
  from quantize import WeightOnlyInt4Linear
label_desc: Use image node_id 4 for calling a global function with example usage: _get_world_size() and returns: int