diff options
Diffstat (limited to 'candle-examples/examples/stable-diffusion/resnet.rs')
| -rw-r--r-- | candle-examples/examples/stable-diffusion/resnet.rs | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/candle-examples/examples/stable-diffusion/resnet.rs b/candle-examples/examples/stable-diffusion/resnet.rs index 5f6a2558..4cfd386d 100644 --- a/candle-examples/examples/stable-diffusion/resnet.rs +++ b/candle-examples/examples/stable-diffusion/resnet.rs @@ -66,6 +66,7 @@ impl ResnetBlock2D { stride: 1, padding: 1, groups: 1, + dilation: 1, }; let norm1 = nn::group_norm(config.groups, in_channels, config.eps, vs.pp("norm1"))?; let conv1 = conv2d(in_channels, out_channels, 3, conv_cfg, vs.pp("conv1"))?; @@ -80,6 +81,7 @@ impl ResnetBlock2D { stride: 1, padding: 0, groups: 1, + dilation: 1, }; Some(conv2d( in_channels, |