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 94f436c8..172a9359 100644 --- a/candle-examples/examples/stable-diffusion/resnet.rs +++ b/candle-examples/examples/stable-diffusion/resnet.rs @@ -66,6 +66,7 @@ impl ResnetBlock2D { let conv_cfg = nn::Conv2dConfig { stride: 1, padding: 1, + groups: 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"))?; @@ -79,6 +80,7 @@ impl ResnetBlock2D { let conv_cfg = nn::Conv2dConfig { stride: 1, padding: 0, + groups: 1, }; Some(conv2d( in_channels, |