diff options
Diffstat (limited to 'candle-wasm-examples/llama2-c')
| -rw-r--r-- | candle-wasm-examples/llama2-c/src/model.rs | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/candle-wasm-examples/llama2-c/src/model.rs b/candle-wasm-examples/llama2-c/src/model.rs index 7471938a..ab9333d2 100644 --- a/candle-wasm-examples/llama2-c/src/model.rs +++ b/candle-wasm-examples/llama2-c/src/model.rs @@ -1,5 +1,7 @@ use candle::{DType, Device, IndexOp, Result, Tensor, D}; -use candle_nn::{embedding, linear, rms_norm, Embedding, Linear, Module, RmsNorm, VarBuilder}; +use candle_nn::{ + embedding, linear_no_bias as linear, rms_norm, Embedding, Linear, Module, RmsNorm, VarBuilder, +}; use std::collections::HashMap; use std::sync::{Arc, Mutex}; |