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;
16pub(crate) use fp8_lut::dequant_nvfp4_to_bf16;
17#[path = "weight_map/loaders_fp8.rs"]
18mod loaders_fp8;
19#[path = "weight_map/loaders_moe.rs"]
20mod loaders_moe;
21#[path = "weight_map/loaders_mtp.rs"]
22mod loaders_mtp;
23#[path = "weight_map/model_a.rs"]
24mod model_a;
25#[path = "weight_map/model_b.rs"]
26mod model_b;
27#[path = "weight_map/moe.rs"]
28mod moe;
29#[path = "weight_map/nemotron.rs"]
30mod nemotron;
31#[path = "weight_map/nvfp4_detect.rs"]
32mod nvfp4_detect;
33#[path = "weight_map/quant_helpers.rs"]
34mod quant_helpers;
35#[path = "weight_map/quantize_fns.rs"]
36mod quantize_fns;
37#[path = "weight_map/quantize_fp8_bs.rs"]
38mod quantize_fp8_bs;
39#[path = "weight_map/quantized.rs"]
40mod quantized;
41#[path = "weight_map/ssm_qwen35.rs"]
42mod ssm_qwen35;
43#[path = "weight_map/ssm_qwen35_more.rs"]
44mod ssm_qwen35_more;
45
46pub use expert::*;
47pub(crate) use fp8_dequant::*;
48pub use loaders_fp8::*;
49pub use loaders_mtp::*;
50pub use model_a::*;
51pub use moe::*;
52pub use nemotron::*;
53pub use nvfp4_detect::*;
54pub use quantize_fns::*;
55pub use quantize_fp8_bs::*;
56pub use quantized::*;
57pub use ssm_qwen35::*;
58
59// Modules whose only exports are `pub(crate)` / `pub(super)` helpers.
60#[allow(unused_imports)]
61pub(crate) use {fp8_lut::*, loaders_moe::*, model_b::*, quant_helpers::*, ssm_qwen35_more::*};
spark_model/weight_map.rs

spark_model/
weight_map.rs
