| 23 | |
| 24 | |
| 25 | def parse_args() -> argparse.Namespace: |
| 26 | parser = argparse.ArgumentParser(description="Python reference MioCodec warmbench.") |
| 27 | parser.add_argument("--model", type=Path, default=DEFAULT_MODEL) |
| 28 | parser.add_argument("--audio", type=Path, default=DEFAULT_AUDIO) |
| 29 | parser.add_argument("--warmup-audio", type=Path, default=DEFAULT_WARMUP_AUDIO) |
| 30 | parser.add_argument("--audio-sequence", default="") |
| 31 | parser.add_argument("--request-json", default="") |
| 32 | parser.add_argument("--request-sequence-json", default="") |
| 33 | parser.add_argument("--warmup-request-json", default="") |
| 34 | parser.add_argument("--backend", choices=("cpu", "cuda"), default="cpu") |
| 35 | parser.add_argument("--device", type=int, default=0) |
| 36 | parser.add_argument("--threads", type=int, default=8) |
| 37 | parser.add_argument("--warmup", type=int, default=1) |
| 38 | parser.add_argument("--iterations", type=int, default=1) |
| 39 | parser.add_argument("--max-audio-seconds", type=float, default=0.0) |
| 40 | parser.add_argument("--audio-out", type=Path, default=Path("miocodec_python_audio.wav")) |
| 41 | parser.add_argument("--audio-out-dir", type=Path, default=None) |
| 42 | parser.add_argument("--output-dir", type=Path, default=None) |
| 43 | parser.add_argument("--timing-file", type=Path, default=Path("miocodec_python_timing.log")) |
| 44 | parser.add_argument("--ssl-checkpoint", type=Path, default=None) |
| 45 | parser.add_argument("--precision", choices=("fp32", "reference"), default="fp32") |
| 46 | return parser.parse_args() |
| 47 | |
| 48 | |
| 49 | def resolve_repo_path(path: Path) -> Path: |