Add shared CLI arguments to an argument parser.
(parser)
| 614 | |
| 615 | |
| 616 | def add_common_args(parser): |
| 617 | """Add shared CLI arguments to an argument parser.""" |
| 618 | parser.add_argument("--csv_file", type=str, default="data/test_prompts_en.csv", |
| 619 | help="CSV file containing prompts and checkpoints") |
| 620 | parser.add_argument("--num_processes", type=int, default=20, |
| 621 | help="Number of parallel worker processes (default: 20)") |
| 622 | parser.add_argument("--images_per_prompt", type=int, default=4, |
| 623 | help="Number of generated images per prompt (default: 4)") |
| 624 | parser.add_argument("--image_suffix", type=str, default=".png", |
| 625 | help="Image file extension (default: .png)") |
| 626 | parser.add_argument("--max_retries", type=int, default=10, |
| 627 | help="Maximum retry attempts per evaluation (default: 10)") |
| 628 | parser.add_argument("--resume", action="store_true", |
| 629 | help="Resume from previous progress instead of starting fresh") |
| 630 | parser.add_argument("--category", type=str, default=None, |
| 631 | help="Category label for output naming (e.g. en, en_long, zh, zh_long)") |
no outgoing calls
no test coverage detected