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: `W2A4`
 - 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 @@
-3a0b9d9543ca72600feb8645e7b59961ba254234f52a50c545b4280594d47487  README.md
-4826107b893c41cf7517727cd2e96bf2eee4a59b1afd597624385b4d307ff83e  assets/image_generation_comparison_matrix.webp
-29926a9051050e69e5ad1dee895eb188c19b091987d6a7e470b93f7695eeeca2  benchmark/summary.json
-aea099a0445d60c272a2c45a8c4e88fd4b3cedf7181a4b3600873104eb7a7941  model.safetensors
-e238cc9bbe81a339e7c5ba5ce4ab7f0d83a40b592714e5bc53629425ece81748  model_index.json
-d72050fd3948b189d54d1dc55020b2f2f3f505d842a178a2121d05bebdcd55d5  orbitquant_codebooks.safetensors
-e3993931d417dcb9e9842490371f297944962ed423dbf6ad93da7496088743a8  orbitquant_manifest.json
+1a52efbd377f8254884be98aa7ab1ef92d4fe599dac50fcb412946c4a365add9  README.md
+812f3828b3bcd19c9879f903c6b768ad2a7d4293016518fda916ab6dba663aa3  assets/image_generation_comparison_matrix.webp
+618770414076f33ca4c7aebe0c173c7e419dcf4c76b4082fb1a8af88bfc54fda  benchmark/summary.json
+aa8e04810221025dc262f66632975bdbf8466b341ad19db61ba4ad4f174f402f  model.safetensors
+33663f26f6c41269f71b4300ba1e92d57e426e52128854b079c25fec0e9080c3  model_index.json
+9ee09b3fcc9615269db665df2a76b9a94099aea67fb5c9ce1523f5c53f0e7723  orbitquant_codebooks.safetensors
+643f1d8f45a489dde2521adad560803bc4949ea9f0f9e732deb8e3999bc575f3  orbitquant_manifest.json
 e1724c9394bf83ea05da6b457b8072d2bcb3d5a533dafce916da8deb517d3825  orbitquant_rotations.safetensors
 65ec8afff39a6cf80feee2eac2f06fa66e51cd471a8382f4355a04fd74eda0af  prompts.json
-aba7920e16eaffbe5208d8ce2b3619cbfd55430487a83c9677a5da08352c750b  quantization_config.json
+6cac0dabb20f4fa9cceda63aa86ba4ca5df66902aeec52b98ab133f0c381bc71  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": 4,
+  "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.39361240202561,
-  "orbitquant_seconds": 0.9436650313436985,
-  "adaln_seconds": 0.011932591442018747,
-  "device_transfer_seconds": 1.4291342729702592,
+  "quantization_elapsed_seconds": 4.982273037079722,
+  "orbitquant_seconds": 3.547152883838862,
+  "adaln_seconds": 0.024637432768940926,
+  "device_transfer_seconds": 1.3974662092514336,
   "module_device_transfer_count": 270,
-  "metrics": {
-    "orbitquant": {
-      "records": 1,
-      "latest_metrics": {
-        "generated_samples": 1,
-        "wall_time_seconds": 35.89397249696776,
-        "peak_vram_bytes": 13017512448
-      }
-    },
-    "original": {
-      "records": 1,
-      "latest_metrics": {
-        "generated_samples": 1,
-        "wall_time_seconds": 5.200292671099305,
-        "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": 4,
+  "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:d72050fd3948b189d54d1dc55020b2f2f3f505d842a178a2121d05bebdcd55d5
+oid sha256:9ee09b3fcc9615269db665df2a76b9a94099aea67fb5c9ce1523f5c53f0e7723
 size 952

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": "4826107b893c41cf7517727cd2e96bf2eee4a59b1afd597624385b4d307ff83e",
-    "benchmark/summary.json": "29926a9051050e69e5ad1dee895eb188c19b091987d6a7e470b93f7695eeeca2",
-    "model.safetensors": "aea099a0445d60c272a2c45a8c4e88fd4b3cedf7181a4b3600873104eb7a7941",
-    "model_index.json": "e238cc9bbe81a339e7c5ba5ce4ab7f0d83a40b592714e5bc53629425ece81748",
-    "orbitquant_codebooks.safetensors": "d72050fd3948b189d54d1dc55020b2f2f3f505d842a178a2121d05bebdcd55d5",
+    "assets/image_generation_comparison_matrix.webp": "812f3828b3bcd19c9879f903c6b768ad2a7d4293016518fda916ab6dba663aa3",
+    "benchmark/summary.json": "618770414076f33ca4c7aebe0c173c7e419dcf4c76b4082fb1a8af88bfc54fda",
+    "model.safetensors": "aa8e04810221025dc262f66632975bdbf8466b341ad19db61ba4ad4f174f402f",
+    "model_index.json": "33663f26f6c41269f71b4300ba1e92d57e426e52128854b079c25fec0e9080c3",
+    "orbitquant_codebooks.safetensors": "9ee09b3fcc9615269db665df2a76b9a94099aea67fb5c9ce1523f5c53f0e7723",
     "orbitquant_rotations.safetensors": "e1724c9394bf83ea05da6b457b8072d2bcb3d5a533dafce916da8deb517d3825",
     "prompts.json": "65ec8afff39a6cf80feee2eac2f06fa66e51cd471a8382f4355a04fd74eda0af",
-    "quantization_config.json": "aba7920e16eaffbe5208d8ce2b3619cbfd55430487a83c9677a5da08352c750b"
+    "quantization_config.json": "6cac0dabb20f4fa9cceda63aa86ba4ca5df66902aeec52b98ab133f0c381bc71"
   }
 }

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
 }