diff options
Diffstat (limited to 'candle-examples/examples/stable-diffusion/vae.rs')
| -rw-r--r-- | candle-examples/examples/stable-diffusion/vae.rs | 4 |
1 files changed, 2 insertions, 2 deletions
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, |