| 148 | shutil.move(str(tmp_log), str(timestamped_dirs[-1] / "sweep.log")) |
| 149 | |
| 150 | modal.Volume.from_name("fused-mm-sample").commit() |
| 151 | |
| 152 | |
| 153 | @app.local_entrypoint() |
| 154 | def main(): |
| 155 | params_file = "quick-bench-params.json" if cfg.sweep == "quick" else "bench-params.json" |
| 156 | bench_params_json = (_bench_params_dir / params_file).read_text() |
| 157 | args = Args( |
| 158 | model=cfg.model, |
| 159 | sweep=cfg.sweep, |
| 160 | tgt_dir=cfg.tgt_dir, |
| 161 | variants=cfg.variants, |
| 162 | bench_params_json=bench_params_json, |
| 163 | resume_experiment=cfg.resume_experiment, |
| 164 | tensor_parallel_size=cfg.n_procs, |
| 165 | ) |