From aba1e90797e430f28eec13b14b76dd5355876f9c Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Wed, 23 Aug 2023 12:58:55 +0100
Subject: 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.
---
 candle-examples/examples/stable-diffusion/vae.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'candle-examples/examples/stable-diffusion/vae.rs')

diff --git a/candle-examples/examples/stable-diffusion/vae.rs b/candle-examples/examples/stable-diffusion/vae.rs
index abba39fa..aa8e13a0 100644
--- a/candle-examples/examples/stable-diffusion/vae.rs
+++ b/candle-examples/examples/stable-diffusion/vae.rs
@@ -51,8 +51,8 @@ impl Encoder {
         config: EncoderConfig,
     ) -> Result<Self> {
         let conv_cfg = nn::Conv2dConfig {
-            stride: 1,
             padding: 1,
+            ..Default::default()
         };
         let conv_in = nn::conv2d(
             in_channels,
@@ -182,8 +182,8 @@ impl Decoder {
         let n_block_out_channels = config.block_out_channels.len();
         let last_block_out_channels = *config.block_out_channels.last().unwrap();
         let conv_cfg = nn::Conv2dConfig {
-            stride: 1,
             padding: 1,
+            ..Default::default()
         };
         let conv_in = nn::conv2d(
             in_channels,
-- 
cgit v1.2.3