surgery v2 (public mirror for colab)

artifacts/m15_scripts/r12_surgery.py

@@ -0,0 +1,71 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""r12_surgery.py — f16 元数据手术 v2 路线 (坑#58: 自写 walk 定位 + GGUFReader 复验双保险)
+用法: python3 r12_surgery.py <f16_path>
+补丁: qwen35.nextn_predict_layers -> 0, qwen35.block_count -> 32
+产物标记: R12_SURGERY_V2_DONE"""
+import struct, json, hashlib, sys
+
+PATH = sys.argv[1] if len(sys.argv) > 1 else "/tmp/k8b/m13_r10_f16.gguf"
+SZT = {0: 1, 1: 1, 2: 2, 3: 2, 4: 4, 5: 4, 6: 4, 7: 1, 10: 8, 11: 8, 12: 8}
+FMT = {0: "<B", 1: "<b", 2: "<H", 3: "<h", 4: "<I", 5: "<i", 6: "<f", 7: "<B", 10: "<Q", 11: "<q", 12: "<d"}
+T_STR, T_ARR = 8, 9
+out = {"path": PATH}
+
+def walk(fh):
+    fh.seek(0)
+    assert fh.read(4) == b"GGUF"
+    ver, n_ten, n_kv = struct.unpack("<IQQ", fh.read(20))
+    idx = {}
+    for _ in range(n_kv):
+        n = struct.unpack("<Q", fh.read(8))[0]
+        key = fh.read(n).decode("utf-8", "replace")
+        vtype = struct.unpack("<I", fh.read(4))[0]
+        if vtype == T_STR:
+            n2 = struct.unpack("<Q", fh.read(8))[0]
+            fh.seek(n2, 1); continue
+        if vtype == T_ARR:
+            etype = struct.unpack("<I", fh.read(4))[0]
+            cnt_off = fh.tell()
+            cnt = struct.unpack("<Q", fh.read(8))[0]
+            if etype == T_STR:
+                for _ in range(cnt):
+                    ln = struct.unpack("<Q", fh.read(8))[0]
+                    fh.seek(ln, 1)
+            else:
+                fh.seek(cnt * SZT[etype], 1)
+            idx[key] = (vtype, etype, cnt, cnt_off, fh.tell() if etype != T_STR else cnt_off)
+            continue
+        fh.seek(SZT[vtype], 1)
+        idx[key] = (vtype, None, None, None, fh.tell() - SZT[vtype])
+    return idx
+
+fh = open(PATH, "r+b")
+idx = walk(fh)
+out["bc_found"] = "qwen35.block_count" in idx
+out["nx_found"] = "qwen35.nextn_predict_layers" in idx
+missing = [k for k in ("qwen35.block_count", "qwen35.nextn_predict_layers") if k not in idx]
+if missing:
+    out["R12_SURGERY_V2_DONE"] = False
+    out["error"] = "keys absent: %s" % missing
+    print(json.dumps(out, ensure_ascii=False)); sys.exit(1)
+for key, newv in (("qwen35.nextn_predict_layers", 0), ("qwen35.block_count", 32)):
+    vtype, _, _, _, off = idx[key]
+    fh.seek(off)
+    old = struct.unpack(FMT[vtype], fh.read(4))[0]
+    fh.seek(off)
+    fh.write(struct.pack(FMT[vtype], newv))
+    out["patch_" + key] = "%s->%s" % (old, newv)
+fh.flush()
+import os
+os.fsync(fh.fileno())
+fh.close()
+
+import gguf
+r = gguf.GGUFReader(PATH)
+bcv = int(r.fields["qwen35.block_count"].parts[-1][0])
+nxv = int(r.fields["qwen35.nextn_predict_layers"].parts[-1][0])
+out["verify"] = "bc=%d nx=%d tensors=%d" % (bcv, nxv, len(r.tensors))
+ok = bcv == 32 and nxv == 0 and len(r.tensors) > 400
+out["R12_SURGERY_V2_DONE"] = bool(ok)
+print(json.dumps(out, ensure_ascii=False))