diff options
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) } |