Add Inkling model support (#31681)

Co-authored-by: Chunan Zeng <zcnrex@gmail.com> Co-authored-by: Ke Bao <ispobaoke@gmail.com> Co-authored-by: Yanbin Jiang <jybsuper@gmail.com> Co-authored-by: Yuhao Yang <47235274+yhyang201@users.noreply.github.com> Co-authored-by: Qiaolin Yu <qiaolin.yu@radixark.ai> Co-authored-by: Zhichen Zeng <zczeng@uw.edu> Co-authored-by: Aurick Qiao <aurick@thinkingmachines.ai> Co-authored-by: Joseph <jk@thinkingmachines.ai>
2026-07-19 22:57:37 -07:00
parent 829e9ce9d5
commit 02236fa38c
279 changed files with 74334 additions and 931 deletions
@@ -0,0 +1,40 @@
+import io
+import os
+import sys
+
+import numpy as np
+import torch
+from PIL import Image
+
+sys.path.insert(0, os.path.dirname(__file__))
+from bench_parity import PS, make_photo_like, ref_patchify
+
+OUT = (
+    sys.argv[1]
+    if len(sys.argv) > 1
+    else os.path.join(os.path.dirname(__file__), "..", "tests", "golden")
+)
+
+CASES = [
+    ("480x640", 480, 640, 10),
+    ("200x320", 200, 320, 11),
+    ("37x53", 37, 53, 12),
+    ("40x40", 40, 40, 13),
+]
+
+os.makedirs(OUT, exist_ok=True)
+for name, h, w, seed in CASES:
+    arr = make_photo_like(h, w, seed=seed)
+    bits = ref_patchify(arr).view(torch.uint16).numpy()
+    buf = io.BytesIO()
+    Image.fromarray(arr).save(buf, format="PNG")
+    path = os.path.join(OUT, f"golden_{name}.npz")
+    np.savez_compressed(
+        path,
+        arr=arr,
+        bits=bits,
+        png=np.frombuffer(buf.getvalue(), dtype=np.uint8),
+        patch_size=np.int64(PS),
+    )
+    print(f"  {path}: input {h}x{w}, bits {bits.shape}")
+print("GOLDEN_OK")