pub fn execute(
plan: &ModelPlan,
weights: &ReferenceWeights,
token_ids: &[u32],
) -> Result<ReferenceOutput, ReferenceError>pub fn execute(
plan: &ModelPlan,
weights: &ReferenceWeights,
token_ids: &[u32],
) -> Result<ReferenceOutput, ReferenceError>