Skip to content

Commit 6e243c2

Browse files
Ronald Tseronaldtse
authored andcommitted
fix: same split for margin_model's precision loop
1 parent 1faedc4 commit 6e243c2

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/gpu/modal_export.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -315,7 +315,7 @@ def stage(event: str) -> None:
315315
meta_path = Path("/root/interscript-ml", spec["metadata"])
316316
mid = re.search(r"^id:\s*(\S+)", meta_path.read_text(encoding="utf-8"), re.M).group(1)
317317
reports: dict[str, str] = {}
318-
for precision in precisions:
318+
for precision in [q.strip() for q in precisions.split(",") if q.strip()]:
319319
zip_path = out_dir / f"{mid}-{precision}.zip"
320320
stage(f"onnx decode {precision}")
321321
report = run_parity(model, zip_path, pairs, max_len=128, reference=reference)

0 commit comments

Comments
 (0)