Method decode

cake-core/src/models/sd/vae.rs:124–135

(
        forwarder: &mut Box<dyn Forwarder>,
        latents: Tensor,
        ctx: &mut Context,
    )

    pub const LAYER_NAME: &'static str = "vae";

    pub async fn decode(
        forwarder: &mut Box<dyn Forwarder>,
        latents: Tensor,
        ctx: &mut Context,
    ) -> anyhow::Result<Tensor> {
        let tensors = Vec::from([Tensor::from_slice(&[0f32], 1, &ctx.device)?, latents]);

        let combined_tensor = pack_tensors(tensors, &ctx.device)?;

        let result = forwarder.forward_mut(&combined_tensor, 0, 0, ctx).await?;
        Ok(result)
    }
}

#[cfg(test)]