code: def add_text_generate_args(parser):
      group = parser.add_argument_group(title="text generation")
      group.add_argument(
          "--port",
          type=int,
          default=5000,
          help="port for text generation server to run on",
      )
      return parser
  
file_path: megatron-lm/tools/run_text_generation_server.py
imports: import os
  import sys
  import socket
  from megatron import get_args
  from megatron import print_rank_0
  from megatron.core import mpu
  from megatron.checkpointing import load_checkpoint
  from megatron.initialize import initialize_megatron
  from megatron.model import GPTModel
  from megatron.training import get_model
  from megatron.arguments import core_transformer_config_from_args
  from megatron.text_generation_server import MegatronServer
  from megatron.text_generation import generate_and_post_process
  from megatron.text_generation import beam_search_and_post_process
  import torch
label_desc: Use image node_id 2 for calling a global function with example usage: add_text_generate_args(parser) and returns: parser