Upload 3 files

segmentation.py

@@ -239,8 +239,10 @@ def q3k_proxy(xg: np.ndarray, wg: np.ndarray | None) -> np.ndarray:
         wu = np.ones_like(u)
     else:
         wu = np.asarray(wg, dtype=np.float64).reshape(n * (B // 16), 16)
-    scale, xmin, codes = make_qkx2_batch(u, wu, nmax=7)
-    deq = scale[:, None] * codes + xmin[:, None]
+    # make_qkx2_batch returns (scale, -min, codes) with the GGML positive-dmin
+    # convention; reconstruction is scale * code - (-min).
+    scale, neg_min, codes = make_qkx2_batch(u, wu, nmax=7)
+    deq = scale[:, None] * codes - neg_min[:, None]
     return deq.reshape(n, B)