MCPcopy Create free account
hub / github.com/evilsocket/cake / encode

Method encode

cake-core/src/models/sd/vae.rs:108–118  ·  view source on GitHub ↗
(
        forwarder: &mut Box<dyn Forwarder>,
        image: Tensor,
        ctx: &mut Context,
    )

Source from the content-addressed store, hash-verified

106 }
107
108 pub async fn encode(
109 forwarder: &mut Box<dyn Forwarder>,
110 image: Tensor,
111 ctx: &mut Context,
112 ) -> anyhow::Result<Tensor> {
113 let tensors = Vec::from([Tensor::from_slice(&[1f32], 1, &ctx.device)?, image]);
114
115 let combined_tensor = pack_tensors(tensors, &ctx.device)?;
116
117 forwarder.forward_mut(&combined_tensor, 0, 0, ctx).await
118 }
119
120 /// The constant layer name for this component.
121 #[cfg(test)]

Callers 7

run_vibevoice_0_5bFunction · 0.45
run_vibevoice_1_5bFunction · 0.45
generate_imageFunction · 0.45
generate_image_openaiFunction · 0.45
text_embeddingsMethod · 0.45
forwardMethod · 0.45
prepare_promptMethod · 0.45

Calls 2

pack_tensorsFunction · 0.85
forward_mutMethod · 0.45

Tested by

no test coverage detected