Skip to main content

load_dense_mlp

Function load_dense_mlp 

Source
pub fn load_dense_mlp(
    shard: &ShardedSafetensors,
    prefix: &str,
    hidden_dim: usize,
    intermediate_dim: usize,
) -> Result<DenseMlpWeights, KimiLoadError>
Expand description

Loads the dense leading layer’s feed-forward block (real tensor names under {prefix}.mlp.*) – Kimi K3’s layer 0 only (first_k_dense_replace=1).