summaryrefslogtreecommitdiff
path: root/candle-examples/examples/stable-diffusion/clip.rs
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2023-08-09 14:23:59 +0200
committerGitHub <noreply@github.com>2023-08-09 13:23:59 +0100
commitbe21d7e75a38b2d383ae64f028ded2d4ce3c473d (patch)
tree5f477f79f8ebcb820cfac34b2cbcf2a8d34dff81 /candle-examples/examples/stable-diffusion/clip.rs
parent9c4cf6804b84ce43c20e04f8737b90a49464367d (diff)
downloadcandle-be21d7e75a38b2d383ae64f028ded2d4ce3c473d.tar.gz
candle-be21d7e75a38b2d383ae64f028ded2d4ce3c473d.tar.bz2
candle-be21d7e75a38b2d383ae64f028ded2d4ce3c473d.zip
Fix the padding used in stable diffusion. (#362)
Diffstat (limited to 'candle-examples/examples/stable-diffusion/clip.rs')
-rw-r--r--candle-examples/examples/stable-diffusion/clip.rs4
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-examples/examples/stable-diffusion/clip.rs b/candle-examples/examples/stable-diffusion/clip.rs
index 12f482fd..ca00b417 100644
--- a/candle-examples/examples/stable-diffusion/clip.rs
+++ b/candle-examples/examples/stable-diffusion/clip.rs
@@ -31,7 +31,7 @@ pub struct Config {
intermediate_size: usize,
pub max_position_embeddings: usize,
// The character to use for padding, use EOS when not set.
- pad_with: Option<String>,
+ pub pad_with: Option<String>,
num_hidden_layers: usize,
num_attention_heads: usize,
#[allow(dead_code)]
@@ -257,7 +257,7 @@ impl ClipEncoder {
fn forward(&self, xs: &Tensor, causal_attention_mask: &Tensor) -> Result<Tensor> {
let mut xs = xs.clone();
for layer in self.layers.iter() {
- xs = layer.forward(&xs, causal_attention_mask)?
+ xs = layer.forward(&xs, causal_attention_mask)?;
}
Ok(xs)
}