Replace artifact with paper-correct codebook v2 and native proof

README.md

@@ -90,7 +90,7 @@ The compact benchmark summary records native BF16-vs-OrbitQuant evidence for the
 - Method: `orbitquant`
 - Bits: `W2A3`
 - Runtime mode: `auto_fused`
-- Activation kernel backend: `triton_cuda`
+- Activation kernel backend: `auto`
 - Activation normalization epsilon: `1e-10`
 - Quantization device: `cuda`
 - Weight quantization backend: `triton_cuda`
@@ -103,7 +103,7 @@ The compact benchmark summary records native BF16-vs-OrbitQuant evidence for the
 - Block size: `paper`
 - Block size policy: `largest_power_of_two_dividing_dim`
 - Codebook: `lloyd_max`
-- Codebook version: `1`
+- Codebook version: `2`
 - Quantized transformer modules: `238`
 - AdaLN INT4 modules: `32`
 - Skipped modules: `6`

SHA256SUMS

@@ -1,10 +1,10 @@
-0ccb2cbee8bf87fff9b604ae7db0eb9e9a821d0cfc21ceeef0ded4bb6107b374  README.md
-87f719a534ba76c19031c82ca761c2fea4e40615f0d1a23f37dd29556ee102aa  assets/image_generation_comparison_matrix.webp
-2bda4afcd7266fbe4e0d184690605dd63f9546fc3a1ec0ca88e117387bf6bdbe  benchmark/summary.json
-aea099a0445d60c272a2c45a8c4e88fd4b3cedf7181a4b3600873104eb7a7941  model.safetensors
-54320a7d5779945db1e5e50b43c11dea40ec5dff74c9dd31a419bda6419160bd  model_index.json
-04da03ff11a91f322d72c88fe5686e7ebf66972de6ed5ab606e5282e682a332f  orbitquant_codebooks.safetensors
-cdddf9d2aa7c409cd59d32d14c6d037b4336706bb9f422b45e1a069bda9127fc  orbitquant_manifest.json
+18ac533ceb7b92c0e1b6ae51376dc405ee7a20981143cc257083bb09bd86bdfc  README.md
+f7e324ad238c63d0d25568acb0d4cee76bf9ddfd3c43f22845ad2138728daef7  assets/image_generation_comparison_matrix.webp
+59f01ffd627235229bc83868ac999ce280ccca656a138603baea85ea32929075  benchmark/summary.json
+aa8e04810221025dc262f66632975bdbf8466b341ad19db61ba4ad4f174f402f  model.safetensors
+61be70974b031f683d3af31c38322999f93c0788909aa7e33b91499962db225a  model_index.json
+3c06a807bddc6af0246586ded2276ec8dca0e309543a9701de1b6872fb8b6b31  orbitquant_codebooks.safetensors
+d0aa18e68101aa217b3bcba80382ebca0ef1a7919b000da5f6f9512e535c41b9  orbitquant_manifest.json
 e1724c9394bf83ea05da6b457b8072d2bcb3d5a533dafce916da8deb517d3825  orbitquant_rotations.safetensors
 65ec8afff39a6cf80feee2eac2f06fa66e51cd471a8382f4355a04fd74eda0af  prompts.json
-a0320d9f96882ee890452ea222e0d4c36db3d5ad57382fc8908a70251d91f614  quantization_config.json
+d5fbd286ff541dc629e0a2b5a216580674faaf7a3814679b0fb7e4a2122cc67b  quantization_config.json

benchmark/summary.json

@@ -1,42 +1,27 @@
 {
-  "status": "metrics_recorded",
+  "status": "not_run",
   "source_model_id": "Tongyi-MAI/Z-Image-Turbo",
   "source_revision": "f332072aa78be7aecdf3ee76d5c247082da564a6",
   "weight_bits": 2,
   "activation_bits": 3,
+  "codebook_version": 2,
   "target_policy": "z_image",
   "runtime_mode": "auto_fused",
-  "activation_kernel_backend": "triton_cuda",
+  "activation_kernel_backend": "auto",
   "activation_eps": 1e-10,
   "quantization_device": "cuda",
   "weight_quantization_backend": "triton_cuda",
   "quantization_staging_mode": "streaming",
   "synchronize_per_module": false,
-  "quantization_elapsed_seconds": 2.160409632138908,
-  "orbitquant_seconds": 0.7088581034913659,
-  "adaln_seconds": 0.011828072369098663,
-  "device_transfer_seconds": 1.4283196232281625,
+  "quantization_elapsed_seconds": 3.5582019980065525,
+  "orbitquant_seconds": 2.1125443908385932,
+  "adaln_seconds": 0.015195954591035843,
+  "device_transfer_seconds": 1.4178410354070365,
   "module_device_transfer_count": 270,
-  "metrics": {
-    "orbitquant": {
-      "records": 1,
-      "latest_metrics": {
-        "generated_samples": 1,
-        "wall_time_seconds": 34.435127603821456,
-        "peak_vram_bytes": 13014530560
-      }
-    },
-    "original": {
-      "records": 1,
-      "latest_metrics": {
-        "generated_samples": 1,
-        "wall_time_seconds": 5.203367706853896,
-        "peak_vram_bytes": 23267081728
-      }
-    }
-  },
+  "metrics": {},
   "native_smoke": {
     "proof_format": "orbitquant-native-smoke-v1",
+    "proof_source": "local_compare_native_bundle",
     "comparison_asset_path": "assets/image_generation_comparison_matrix.webp",
     "paired_prompt_seed_count": 1,
     "paired_prompt_seed_keys": [

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:aea099a0445d60c272a2c45a8c4e88fd4b3cedf7181a4b3600873104eb7a7941
+oid sha256:aa8e04810221025dc262f66632975bdbf8466b341ad19db61ba4ad4f174f402f
 size 1599432256

model_index.json

@@ -13,9 +13,10 @@
   "rotations": "orbitquant_rotations.safetensors",
   "weight_bits": 2,
   "activation_bits": 3,
+  "codebook_version": 2,
   "target_policy": "z_image",
   "runtime_mode": "auto_fused",
-  "activation_kernel_backend": "triton_cuda",
+  "activation_kernel_backend": "auto",
   "activation_eps": 1e-10,
   "quantization_device": "cuda",
   "weight_quantization_backend": "triton_cuda",

orbitquant_codebooks.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:04da03ff11a91f322d72c88fe5686e7ebf66972de6ed5ab606e5282e682a332f
+oid sha256:3c06a807bddc6af0246586ded2276ec8dca0e309543a9701de1b6872fb8b6b31
 size 816

orbitquant_manifest.json

@@ -12,10 +12,10 @@
   "block_size": "paper",
   "block_size_policy": "largest_power_of_two_dividing_dim",
   "codebook": "lloyd_max",
-  "codebook_version": 1,
+  "codebook_version": 2,
   "row_norm_dtype": "bfloat16",
   "runtime_mode": "auto_fused",
-  "activation_kernel_backend": "triton_cuda",
+  "activation_kernel_backend": "auto",
   "activation_eps": 1e-10,
   "adaln_group_size": 64,
   "quantization_device": "cuda",
@@ -2721,13 +2721,13 @@
     ]
   },
   "checksums": {
-    "assets/image_generation_comparison_matrix.webp": "87f719a534ba76c19031c82ca761c2fea4e40615f0d1a23f37dd29556ee102aa",
-    "benchmark/summary.json": "2bda4afcd7266fbe4e0d184690605dd63f9546fc3a1ec0ca88e117387bf6bdbe",
-    "model.safetensors": "aea099a0445d60c272a2c45a8c4e88fd4b3cedf7181a4b3600873104eb7a7941",
-    "model_index.json": "54320a7d5779945db1e5e50b43c11dea40ec5dff74c9dd31a419bda6419160bd",
-    "orbitquant_codebooks.safetensors": "04da03ff11a91f322d72c88fe5686e7ebf66972de6ed5ab606e5282e682a332f",
+    "assets/image_generation_comparison_matrix.webp": "f7e324ad238c63d0d25568acb0d4cee76bf9ddfd3c43f22845ad2138728daef7",
+    "benchmark/summary.json": "59f01ffd627235229bc83868ac999ce280ccca656a138603baea85ea32929075",
+    "model.safetensors": "aa8e04810221025dc262f66632975bdbf8466b341ad19db61ba4ad4f174f402f",
+    "model_index.json": "61be70974b031f683d3af31c38322999f93c0788909aa7e33b91499962db225a",
+    "orbitquant_codebooks.safetensors": "3c06a807bddc6af0246586ded2276ec8dca0e309543a9701de1b6872fb8b6b31",
     "orbitquant_rotations.safetensors": "e1724c9394bf83ea05da6b457b8072d2bcb3d5a533dafce916da8deb517d3825",
     "prompts.json": "65ec8afff39a6cf80feee2eac2f06fa66e51cd471a8382f4355a04fd74eda0af",
-    "quantization_config.json": "a0320d9f96882ee890452ea222e0d4c36db3d5ad57382fc8908a70251d91f614"
+    "quantization_config.json": "d5fbd286ff541dc629e0a2b5a216580674faaf7a3814679b0fb7e4a2122cc67b"
   }
 }

quantization_config.json

@@ -6,6 +6,7 @@
   "rotation_seed": 0,
   "block_size": "paper",
   "codebook": "lloyd_max",
+  "codebook_version": 2,
   "codebook_dtype": "float32",
   "row_norm_dtype": "bfloat16",
   "activation_norm_dtype": "float32",
@@ -18,9 +19,9 @@
   "modules_dtype_dict": {},
   "artifact_format_version": 1,
   "runtime_mode": "auto_fused",
-  "activation_kernel_backend": "triton_cuda",
-  "packed_matmul_block_m": 32,
+  "activation_kernel_backend": "auto",
+  "packed_matmul_block_m": 64,
   "packed_matmul_block_n": 64,
-  "packed_matmul_block_k": 64,
-  "packed_matmul_num_warps": 8
+  "packed_matmul_block_k": 128,
+  "packed_matmul_num_warps": 4
 }