(
forwarder: &mut Box<dyn Forwarder>,
latent_model_input: Tensor,
text_embeddings: Tensor,
timestep: usize,
ctx: &mut Context,
)
| 102 | } |
| 103 | |
| 104 | pub async fn forward_unpacked( |
| 105 | forwarder: &mut Box<dyn Forwarder>, |
| 106 | latent_model_input: Tensor, |
| 107 | text_embeddings: Tensor, |
| 108 | timestep: usize, |
| 109 | ctx: &mut Context, |
| 110 | ) -> anyhow::Result<Tensor> { |
| 111 | // Pack the tensors to be sent into one |
| 112 | let timestep_tensor = Tensor::from_slice(&[timestep as f32], 1, &ctx.device)?; |
| 113 | |
| 114 | let tensors = Vec::from([latent_model_input, text_embeddings, timestep_tensor]); |
| 115 | |
| 116 | let combined_tensor = pack_tensors(tensors, &ctx.device)?; |
| 117 | forwarder.forward_mut(&combined_tensor, 0, 0, ctx).await |
| 118 | } |
| 119 | } |
| 120 | |
| 121 | #[cfg(test)] |
nothing calls this directly
no test coverage detected