huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-20 20:09:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add some missing biases. (#908)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-20 10:14:51 +01:00
committed by GitHub
parent c0b49d5a50
commit f685b2231c
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
candle-transformers/src/models/wuerstchen/paella_vq.rs
View File

@ -106,8 +106,7 @@ impl PaellaVQ {
stride: 2,
..Default::default()
};
let block =
candle_nn::conv2d_no_bias(C_LEVELS[i - 1], c_level, 4, cfg, vb_d.pp(d_idx))?;
let block = candle_nn::conv2d(C_LEVELS[i - 1], c_level, 4, cfg, vb_d.pp(d_idx))?;
d_idx += 1;
Some(block)
} else {
@ -130,7 +129,7 @@ impl PaellaVQ {
let mut up_blocks = Vec::new();
let vb_u = vb.pp("up_blocks");
let mut u_idx = 0;
let up_blocks_conv = candle_nn::conv2d_no_bias(
let up_blocks_conv = candle_nn::conv2d(
LATENT_CHANNELS,
C_LEVELS[1],
1,
@ -152,7 +151,7 @@ impl PaellaVQ {
stride: 2,
..Default::default()
};
let block = candle_nn::conv_transpose2d_no_bias(
let block = candle_nn::conv_transpose2d(
c_level,
C_LEVELS[C_LEVELS.len() - i - 2],
4,