From bf1f4f72f1a8466b38a464041cee6d37ebc2806a Mon Sep 17 00:00:00 2001 From: Ronald Tse Date: Sat, 29 Aug 2026 17:54:41 +0800 Subject: [PATCH] fix: rebuild_int8_head32 shares the resumable reference decode MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit The sweep built four head32 zips then died mid-reference-decode (same traceback-less container kill; no margins written). Same resume file per model dir — parity and rebuild progress compose. --- src/gpu/modal_export.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/gpu/modal_export.py b/src/gpu/modal_export.py index 2e6a2ed..4d28407 100644 --- a/src/gpu/modal_export.py +++ b/src/gpu/modal_export.py @@ -478,7 +478,12 @@ def rebuild_int8_head32(model_id: str, limit: int = 0) -> dict: else: dst.writestr(name, src.read(name)) - reference = reference_decode(model, [s for s, _ in pairs], max_len=128) + reference = reference_decode( + model, + [s for s, _ in pairs], + max_len=128, + resume_path=Path("/outputs/imf") / model_id / "reference_progress.jsonl", + ) report = run_parity(model, new_zip, pairs, max_len=128, reference=reference) if not report.passed: raise RuntimeError(f"parity gate FAILED for {new_zip.name}: {report}")