Method load

cake-core/src/models/luxtts/model.rs:38–41

(dim: usize, vb: VarBuilder)

impl BypassModule {
    fn load(dim: usize, vb: VarBuilder) -> Result<Self> {
        let bypass_scale = vb.get(dim, "bypass_scale")?;
        Ok(Self { bypass_scale })
    }

    fn forward(&self, src_orig: &Tensor, processed: &Tensor) -> Result<Tensor> {
        // Python: output = src_orig + (processed - src_orig) * bypass_scale