diff options
Diffstat (limited to 'candle-examples/examples/musicgen/encodec_model.rs')
| -rw-r--r-- | candle-examples/examples/musicgen/encodec_model.rs | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/candle-examples/examples/musicgen/encodec_model.rs b/candle-examples/examples/musicgen/encodec_model.rs index 9c966497..e7712bf3 100644 --- a/candle-examples/examples/musicgen/encodec_model.rs +++ b/candle-examples/examples/musicgen/encodec_model.rs @@ -274,14 +274,22 @@ impl EncodecConv1d { in_c, out_c, kernel_size, - Conv1dConfig { padding: 0, stride }, + Conv1dConfig { + padding: 0, + stride, + groups: 1, + }, vb.pp("conv"), )?, NormType::None => conv1d( in_c, out_c, kernel_size, - Conv1dConfig { padding: 0, stride }, + Conv1dConfig { + padding: 0, + stride, + groups: 1, + }, vb.pp("conv"), )?, }; |