diff options
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 { |