INNER CODE UNIT · Python

parse_args

invergent-ai/surogate · surogate/cli/main.py:83

def parse_args():
    print_banner(_get_version())

    parser = argparse.ArgumentParser(description="Surogate LLMOps Framework")
    parser.set_defaults(func=lambda _args, p=parser: p.print_help())
    subparsers = parser.add_subparsers(dest="command", metavar="<command>")

    # sft command
    from surogate.cli.sft import prepare_command_parser as sft_prepare_command_parser

    sft_prepare_command_parser(subparsers.add_parser("sft", help="Supervised Fine-Tuning"))

    # dpo command
    from surogate.cli.dpo import prepare_command_parser as dpo_prepare_command_parser

    dpo_prepare_command_parser(subparsers.add_parser("dpo", help="Direct Preference Optimization (offline)"))

    # pretrain command

View source record →