Skip to main content

load_f32_vec

Function load_f32_vec 

Source
pub fn load_f32_vec(
    shard: &ShardedSafetensors,
    name: &str,
) -> Result<Vec<f32>, KimiLoadError>
Expand description

Reads any real tensor as an owned f32 vector, dispatching on its real declared dtype (F32 direct, BF16 dequantized) – exposed pub since not every real weight (e.g. the per-layer input_layernorm.weight/post_attention_layernorm.weight, which aren’t nested inside KdaAttnWeights/MlaAttnWeights/ DenseMlpWeights/BlockResidualWeights) has a dedicated loader function above.