code: def parse_args(
      parser: argparse.ArgumentParser,
  ) -> argparse.ArgumentParser:
      parser.add_argument(
          "--variant",
          type=str,
          required=True,
          help="Provide fastvit model variant name.",
      )
      parser.add_argument(
          "--output-dir",
          type=str,
          default=".",
          help="Provide location to save exported models.",
      )
      parser.add_argument(
          "--checkpoint",
          type=str,
          default=None,
          help="Provide location of trained checkpoint.",
      )
      return parser
  
imports: import argparse
  import os
  import coremltools
  import torch
  import models
  from models.modules.mobileone import reparameterize_model
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args(parser) and returns: parser