Method load

cake-core/src/models/vibevoice/eos_classifier.rs:24–31

(vb: VarBuilder, backend: Arc<dyn ComputeBackend>)

impl EosClassifier {
    pub fn load(vb: VarBuilder, backend: Arc<dyn ComputeBackend>) -> Result<Self> {
        // Weight shapes inferred from safetensors: fc1 + fc2
        let fc1_weight = vb.pp("fc1").get_unchecked("weight")?;
        let fc1_bias = Some(vb.pp("fc1").get_unchecked("bias")?);
        let fc2_weight = vb.pp("fc2").get_unchecked("weight")?;
        let fc2_bias = Some(vb.pp("fc2").get_unchecked("bias")?);
        Ok(Self { fc1_weight, fc1_bias, fc2_weight, fc2_bias, backend })
    }

    /// Predict EOS probability.
    /// Input: (batch, hidden_size) LLM hidden state.