diff options
Diffstat (limited to 'candle-examples/examples/stable-diffusion/clip.rs')
-rw-r--r-- | candle-examples/examples/stable-diffusion/clip.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/stable-diffusion/clip.rs b/candle-examples/examples/stable-diffusion/clip.rs index 86313c52..be798ad0 100644 --- a/candle-examples/examples/stable-diffusion/clip.rs +++ b/candle-examples/examples/stable-diffusion/clip.rs @@ -294,7 +294,7 @@ impl ClipTextTransformer { } impl ClipTextTransformer { - fn forward(&self, xs: &Tensor) -> Result<Tensor> { + pub fn forward(&self, xs: &Tensor) -> Result<Tensor> { let (bsz, seq_len) = xs.dims2()?; let xs = self.embeddings.forward(xs)?; let causal_attention_mask = Self::build_causal_attention_mask(bsz, seq_len, xs.device())?; |