Method adaln_modulate

cake-core/src/backends/cpu/mod.rs:274–285

(
        &self,
        x: &Tensor,
        norm_weight: &Tensor,
        scale: &Tensor,
        shift: &Tensor,
        eps: f32,
    )

    }

    fn adaln_modulate(
        &self,
        x: &Tensor,
        norm_weight: &Tensor,
        scale: &Tensor,
        shift: &Tensor,
        eps: f32,
    ) -> Result<Tensor> {
        // rms_norm(x) * (1 + scale) + shift
        let n = candle_nn::ops::rms_norm(&x.contiguous()?, norm_weight, eps)?;
        n.broadcast_mul(&(scale + 1.0)?)? + shift
    }

    fn sigmoid(&self, x: &Tensor) -> Result<Tensor> {
        if x.dtype() == DType::F32 {