diff options
Diffstat (limited to 'candle-nn/src/lib.rs')
| -rw-r--r-- | candle-nn/src/lib.rs | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/candle-nn/src/lib.rs b/candle-nn/src/lib.rs index eb3cde4a..2113566d 100644 --- a/candle-nn/src/lib.rs +++ b/candle-nn/src/lib.rs @@ -46,7 +46,9 @@ pub use embedding::{embedding, Embedding}; pub use func::{func, func_t, Func, FuncT}; pub use group_norm::{group_norm, GroupNorm}; pub use init::Init; -pub use layer_norm::{layer_norm, rms_norm, LayerNorm, LayerNormConfig, RmsNorm}; +pub use layer_norm::{ + layer_norm, layer_norm_no_bias, rms_norm, LayerNorm, LayerNormConfig, RmsNorm, +}; pub use linear::{linear, linear_b, linear_no_bias, Linear}; pub use ops::Dropout; pub use optim::{AdamW, Optimizer, ParamsAdamW, SGD}; |