diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-29 19:12:16 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-29 19:12:16 +0100 |
| commit | 2d3fcad26788dff3fa73996a3cc8e5fd5382f6b2 (patch) | |
| tree | d2ab2b3a5b0a08903123a9039319f30e9fe6cc07 /candle-examples/examples/stable-diffusion/unet_2d_blocks.rs | |
| parent | b31d41e26a47d91d828e0c4f567f14b659775e5e (diff) | |
| download | candle-2d3fcad26788dff3fa73996a3cc8e5fd5382f6b2.tar.gz candle-2d3fcad26788dff3fa73996a3cc8e5fd5382f6b2.tar.bz2 candle-2d3fcad26788dff3fa73996a3cc8e5fd5382f6b2.zip | |
Simplify usage of the pool functions. (#662)
* Simplify usage of the pool functions.
* Small tweak.
* Attempt at using apply to simplify the convnet definition.
Diffstat (limited to 'candle-examples/examples/stable-diffusion/unet_2d_blocks.rs')
| -rw-r--r-- | candle-examples/examples/stable-diffusion/unet_2d_blocks.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/stable-diffusion/unet_2d_blocks.rs b/candle-examples/examples/stable-diffusion/unet_2d_blocks.rs index 1db65222..26a1035b 100644 --- a/candle-examples/examples/stable-diffusion/unet_2d_blocks.rs +++ b/candle-examples/examples/stable-diffusion/unet_2d_blocks.rs @@ -47,7 +47,7 @@ impl Downsample2D { fn forward(&self, xs: &Tensor) -> Result<Tensor> { let _enter = self.span.enter(); match &self.conv { - None => xs.avg_pool2d((2, 2), (2, 2)), + None => xs.avg_pool2d(2), Some(conv) => { if self.padding == 0 { let xs = xs |