Bugfix for applying the bias in conv1d-transpose. (#1732)

huggingface · Feb 18, 2024 · 3ba3744 · 3ba3744
1 parent 1fb7287
commit 3ba3744
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/candle-nn/src/conv.rs b/candle-nn/src/conv.rs
@@ -134,7 +134,7 @@ impl crate::Module for ConvTranspose1d {
             None => Ok(x),
             Some(bias) => {
                 let b = bias.dims1()?;
-                let bias = bias.reshape((1, b, 1, 1))?;
+                let bias = bias.reshape((1, b, 1))?;
                 Ok(x.broadcast_add(&bias)?)
             }
         }