derive: restore the header read

2026-10-06 21:42:16 -07:00
parent 650dfff5a2
commit 319dd0f7da
1 changed files with 1 additions and 0 deletions
@@ -37,6 +37,7 @@ NAME_RE = re.compile(r"^gram\.r(\d)\.blk\.(\d+)\.ffn_down_exps\.(\d+)\.bin$")
 def read_gram(path):
    """(k, rank, rows, trace) from a RADGRAM2 file (full matrix on disk)."""
    with open(path, "rb") as f:
+        hdr = f.read(32)
        magic, k, rank, rows, domain = struct.unpack("<8sIIQI", hdr[:28])
        if magic != b"RADGRAM2":
            raise SystemExit(f"not a RADGRAM2 file: {path}")