Views
No views yet
nn.Linear layers quantized to bitsandbytes INT8 weight-only format. Patch Conv2d, learned position embedding, RMSNorm and activations remain FP16.| Version | Checkpoint size | Pooled feature cosine vs FP16 |
|---|---|---|
| Original BF16 | 802.45 MB | 1.0 |
| This INT8 W8A16 | 407.37 MB | 0.99617 mean |
INT8_STRESS_REPORT.md and the raw JSON results.moonvit_v2_bnb_int8_w8a16.pt — packed INT8 checkpointload_int8_moonvit.py — standalone loadermoonvit.py — MoonViT-V2 architecturevision_config.json — configurationBUILD_MANIFEST.json — exact size, SHA-256 and smoke testQUANTIZATION_REPORT.md, INT8_STRESS_REPORT.md — evaluation reportsLICENSE — upstream Kimi K3 Licensepip install torch bitsandbytes>=0.501from huggingface_hub import snapshot_download
2import sys
3
4repo_dir = snapshot_download("cwnd/Kimi-K3-MoonViT-V2-INT8-W8A16")
5sys.path.insert(0, repo_dir)
6
7from load_int8_moonvit import load_int8_moonvit
8
9model = load_int8_moonvit(
10 f"{repo_dir}/moonvit_v2_bnb_int8_w8a16.pt",
11 device="cuda",
12)[T,H,W] grid:1outputs = model(pixel_values, grid_thws)
2# For one 224×224 image:
3# outputs[0].shape == [64, 4, 1024]pixel_values has shape [num_patches, 3, 14, 14]. grid_thws contains one row per image, with height and width expressed in patch units.[1,4,1024]BUILD_MANIFEST.json for the checkpoint SHA-256.