def METHOD_NAME(argv):
  """Parses args for the workflow."""
  parser = argparse.ArgumentParser()
  parser.add_argument(
      '--input',
      dest='input',
      help='Path to the text file containing sentences.')
  parser.add_argument(
      '--output',
      dest='output',
      required=True,
      help='Path of file in which to save the output predictions.')
  parser.add_argument(
      '--model_name',
      dest='model_name',
      required=True,
      help='bert uncased model. This can be base model or large model')
  parser.add_argument(
      '--model_class',
      dest='model_class',
      default=AutoModelForMaskedLM,
      help='Name of the model from Hugging Face')
  parser.add_argument(
      '--large_model',
      action='store_true',
      dest='large_model',
      default=False,
      help='Set to true if your model is large enough to run into memory '
      'pressure if you load multiple copies.')
  return parser.METHOD_NAME(argv)