From b9c55ff41479d0d3ab3dd09de782f28f76bb8e5f Mon Sep 17 00:00:00 2001 From: Ronald Tse Date: Mon, 31 Aug 2026 20:40:20 +0200 Subject: [PATCH] =?UTF-8?q?fix(e5):=20build=5Fmtp=20takes=20steps=20only?= =?UTF-8?q?=20=E2=80=94=20beta=20leaked=20in=20via=20the=20spec=20splat?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/gpu/modal_distill.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/gpu/modal_distill.py b/src/gpu/modal_distill.py index ac5a14b..6000018 100644 --- a/src/gpu/modal_distill.py +++ b/src/gpu/modal_distill.py @@ -749,7 +749,8 @@ def distill_sequence(spec_id: str, epochs: int = 3) -> dict: _ensure_src_path() from gpu.mtp import build_mtp - mtp_head = build_mtp(student, **spec["mtp_aux"]) + mtp_cfg = spec["mtp_aux"] + mtp_head = build_mtp(student, steps=int(mtp_cfg.get("steps", 3))) n = sum(q.numel() for q in mtp_head.parameters()) / 1e6 print(f"[{spec_id}] mtp_aux head: {n:.2f}M params", flush=True) student.train()