huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add some group parameter to convolutions. (#566)

Browse Source 
* Add some group parameter to convolutions.

* Avoid some unnecessary groups checks.

* Move the tensor convolution bits.

* Properh handling of groups.

* Bump the crate version.

* And add a changelog.
This commit is contained in:
Laurent Mazare
2023-08-23 12:58:55 +01:00
committed by GitHub
parent 4ee1cf038a
commit aba1e90797
30 changed files with 216 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
candle-examples/examples/musicgen/encodec_model.rs
View File

@ -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"),
)?,
};