Simple pad support. (#336)

* Simple pad support. * Fix the tensor indexing when padding.
2025-06-17 02:58:50 +00:00 · 2023-08-07 16:24:56 +02:00
parent e72ba0b9e7
commit f53a333ea9
4 changed files with 31 additions and 7 deletions
--- a/candle-examples/examples/stable-diffusion/embeddings.rs
+++ b/candle-examples/examples/stable-diffusion/embeddings.rs
@ -57,7 +57,7 @@ impl Timesteps {
            Tensor::cat(&[&sin, &cos], D::Minus1)?
        };
        if self.num_channels % 2 == 1 {
-            crate::utils::pad(&emb) // ([0, 1, 0, 0], 'constant', None)
+            emb.pad_with_zeros(D::Minus2, 0, 1)
        } else {
            Ok(emb)
        }