| 18 | |
| 19 | |
| 20 | def parse_args() -> argparse.Namespace: |
| 21 | parser = argparse.ArgumentParser(description="Deploy a model checkpoint using ART") |
| 22 | |
| 23 | parser.add_argument("--project", required=True, help="ART project name") |
| 24 | parser.add_argument("--model", required=True, help="Name of the model to deploy") |
| 25 | parser.add_argument("--base-model", required=True, help="Base model to use") |
| 26 | |
| 27 | # Optional arguments |
| 28 | parser.add_argument( |
| 29 | "--backup-bucket", |
| 30 | help="Name of the S3 bucket containing model checkpoints", |
| 31 | ) |
| 32 | parser.add_argument( |
| 33 | "--step", |
| 34 | type=str, |
| 35 | default="latest", |
| 36 | help="Training step to deploy (should correspond to a saved checkpoint)", |
| 37 | ) |
| 38 | parser.add_argument( |
| 39 | "--art-path", type=str, help="Path to the ART directory", default=".art" |
| 40 | ) |
| 41 | |
| 42 | return parser.parse_args() |
| 43 | |
| 44 | |
| 45 | # --------------------------------------------------------------------------- |