spark_model/
weight_map.rs

1// SPDX-License-Identifier: AGPL-3.0-only
2
3//! Weight name mapping from HuggingFace safetensors to typed layer structures.
4//!
5//! Maps the 72 unique weight patterns from Qwen3-Next-80B-A3B-Instruct-NVFP4
6//! into structured per-layer weight references.
7//!
8//! Refactor wave 4a (2026-05-03): split into `weight_map/` sub-modules.
9
10#[path = "weight_map/expert.rs"]
11mod expert;
12#[path = "weight_map/fp8_dequant.rs"]
13mod fp8_dequant;
14#[path = "weight_map/fp8_lut.rs"]
15mod fp8_lut;
16#[path = "weight_map/loaders_fp8.rs"]
17mod loaders_fp8;
18#[path = "weight_map/loaders_moe.rs"]
19mod loaders_moe;
20#[path = "weight_map/loaders_mtp.rs"]
21mod loaders_mtp;
22#[path = "weight_map/model_a.rs"]
23mod model_a;
24#[path = "weight_map/model_b.rs"]
25mod model_b;
26#[path = "weight_map/moe.rs"]
27mod moe;
28#[path = "weight_map/nemotron.rs"]
29mod nemotron;
30#[path = "weight_map/nvfp4_detect.rs"]
31mod nvfp4_detect;
32#[path = "weight_map/quant_helpers.rs"]
33mod quant_helpers;
34#[path = "weight_map/quantize_fns.rs"]
35mod quantize_fns;
36#[path = "weight_map/quantize_fp8_bs.rs"]
37mod quantize_fp8_bs;
38#[path = "weight_map/quantized.rs"]
39mod quantized;
40#[path = "weight_map/ssm_qwen35.rs"]
41mod ssm_qwen35;
42#[path = "weight_map/ssm_qwen35_more.rs"]
43mod ssm_qwen35_more;
44
45pub use expert::*;
46pub(crate) use fp8_dequant::*;
47pub use loaders_fp8::*;
48pub use loaders_mtp::*;
49pub use model_a::*;
50pub use moe::*;
51pub use nemotron::*;
52pub use nvfp4_detect::*;
53pub use quantize_fns::*;
54pub use quantize_fp8_bs::*;
55pub use quantized::*;
56pub use ssm_qwen35::*;
57
58// Modules whose only exports are `pub(crate)` / `pub(super)` helpers.
59#[allow(unused_imports)]
60pub(crate) use {fp8_lut::*, loaders_moe::*, model_b::*, quant_helpers::*, ssm_qwen35_more::*};
spark_model/weight_map.rs

spark_model/
weight_map.rs
