diff options
author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-23 12:58:55 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-08-23 12:58:55 +0100 |
commit | aba1e90797e430f28eec13b14b76dd5355876f9c (patch) | |
tree | 16bcf7fb151715d3bcdbec2b5263922bd0bdd35a /candle-examples/examples/stable-diffusion/unet_2d_blocks.rs | |
parent | 4ee1cf038ada55ec477dcd6496cf2aec1902775b (diff) | |
download | candle-aba1e90797e430f28eec13b14b76dd5355876f9c.tar.gz candle-aba1e90797e430f28eec13b14b76dd5355876f9c.tar.bz2 candle-aba1e90797e430f28eec13b14b76dd5355876f9c.zip |
Add some group parameter to convolutions. (#566)
* Add some group parameter to convolutions.
* Avoid some unnecessary groups checks.
* Move the tensor convolution bits.
* Properh handling of groups.
* Bump the crate version.
* And add a changelog.
Diffstat (limited to 'candle-examples/examples/stable-diffusion/unet_2d_blocks.rs')
-rw-r--r-- | candle-examples/examples/stable-diffusion/unet_2d_blocks.rs | 6 |
1 files changed, 5 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 b7adb2c0..65341e74 100644 --- a/candle-examples/examples/stable-diffusion/unet_2d_blocks.rs +++ b/candle-examples/examples/stable-diffusion/unet_2d_blocks.rs @@ -24,7 +24,11 @@ impl Downsample2D { padding: usize, ) -> Result<Self> { let conv = if use_conv { - let config = nn::Conv2dConfig { stride: 2, padding }; + let config = nn::Conv2dConfig { + stride: 2, + padding, + ..Default::default() + }; let conv = conv2d(in_channels, out_channels, 3, config, vs.pp("conv"))?; Some(conv) } else { |