Skip to main content

load_expert_stack

Function load_expert_stack 

Source
pub fn load_expert_stack(
    loader: &mut dyn WeightLoader,
    key: &str,
    num_experts: usize,
    k: usize,
    n: usize,
) -> Result<Vec<f32>>
Expand description

Load [num_experts, k, n] f32 expert stack from a loader, verifying shape.