| 166 | return fig |
| 167 | |
| 168 | |
| 169 | class Args(BaseSettings, cli_parse_args=True): |
| 170 | base_dir: Path = Path(__file__).parent / "modal-results" |
| 171 | gpu: str = "b200" |
| 172 | bench_fn: str = "own" |
| 173 | case: str = "large" |
| 174 | tps: list[int] = [1, 2, 4, 8] |
| 175 | h_values: list[int] = [16, 64, 256] |
| 176 | out: Path = Path(__file__).parent.parent / "imgs" / "tp-scaling" / "tp-scaling.pdf" |
| 177 | use_reruns: bool = False |
| 178 | max_slowdown: float = 2.0 # drop runs whose FMMS median is more than this factor slower than the fastest at the same tp |
| 179 | |
| 180 | |