metrale_model_layers/
weight_map.rs

1// SPDX-License-Identifier: MIT OR Apache-2.0
2
3//! 2026-09-25: Maps HuggingFace safetensors tensor names onto the typed per-layer
4//! weight structures, through the `weight_map/` sub-modules re-exported here.
5//!
6//! Owner: model-layers (weight loading).
7//! Invariants: none beyond the types.
8
9#[path = "weight_map/expert.rs"]
10mod expert;
11#[path = "weight_map/fp8_dequant.rs"]
12pub mod fp8_dequant;
13#[path = "weight_map/fp8_lut.rs"]
14pub mod fp8_lut;
15pub(crate) use fp8_lut::dequant_nvfp4_to_bf16;
16#[path = "weight_map/loaders_fp8.rs"]
17pub mod loaders_fp8;
18#[path = "weight_map/loaders_moe.rs"]
19pub mod loaders_moe;
20#[path = "weight_map/loaders_mtp.rs"]
21mod loaders_mtp;
22#[path = "weight_map/model_a.rs"]
23pub mod 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"]
29pub mod nemotron;
30#[path = "weight_map/nvfp4_detect.rs"]
31pub mod nvfp4_detect;
32#[path = "weight_map/quant_helpers.rs"]
33pub mod 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"]
39pub mod quantized;
40#[path = "weight_map/ssm_qwen35.rs"]
41pub mod ssm_qwen35;
42#[path = "weight_map/ssm_qwen35_more.rs"]
43pub mod 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#[allow(unused_imports)]
59pub(crate) use {fp8_lut::*, loaders_moe::*, model_b::*, quant_helpers::*, ssm_qwen35_more::*};
metrale_model_layers/weight_map.rs

metrale_model_layers/
weight_map.rs
