diff --git a/src/gpu/modal_export.py b/src/gpu/modal_export.py index 9e0b933..1836db1 100644 --- a/src/gpu/modal_export.py +++ b/src/gpu/modal_export.py @@ -315,7 +315,7 @@ def stage(event: str) -> None: meta_path = Path("/root/interscript-ml", spec["metadata"]) mid = re.search(r"^id:\s*(\S+)", meta_path.read_text(encoding="utf-8"), re.M).group(1) reports: dict[str, str] = {} - for precision in precisions: + for precision in [q.strip() for q in precisions.split(",") if q.strip()]: zip_path = out_dir / f"{mid}-{precision}.zip" stage(f"onnx decode {precision}") report = run_parity(model, zip_path, pairs, max_len=128, reference=reference) @@ -382,7 +382,7 @@ def margin_model( meta_path = Path("/root/interscript-ml", spec["metadata"]) mid = re.search(r"^id:\s*(\S+)", meta_path.read_text(encoding="utf-8"), re.M).group(1) reports: dict[str, str] = {} - for precision in precisions: + for precision in [p.strip() for p in precisions.split(",") if p.strip()]: zip_path = out_dir / f"{mid}-{precision}.zip" if not zip_path.exists(): reports[precision] = "zip not exported (skipped)"