diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-07 16:24:56 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-07 15:24:56 +0100 |
| commit | f53a333ea91233b41dd946c2c30213c79b4d1cb3 (patch) | |
| tree | e9e525b6161f7fb31010ac26461f241729b22d75 /candle-examples/examples/stable-diffusion/embeddings.rs | |
| parent | e72ba0b9e755bbac5bd60718765c043bba3a63dc (diff) | |
| download | candle-f53a333ea91233b41dd946c2c30213c79b4d1cb3.tar.gz candle-f53a333ea91233b41dd946c2c30213c79b4d1cb3.tar.bz2 candle-f53a333ea91233b41dd946c2c30213c79b4d1cb3.zip | |
Simple pad support. (#336)
* Simple pad support.
* Fix the tensor indexing when padding.
Diffstat (limited to 'candle-examples/examples/stable-diffusion/embeddings.rs')
| -rw-r--r-- | candle-examples/examples/stable-diffusion/embeddings.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/stable-diffusion/embeddings.rs b/candle-examples/examples/stable-diffusion/embeddings.rs index f8a4f351..848f1760 100644 --- 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) } |