diff options
Diffstat (limited to 'candle-examples/examples/falcon/model.rs')
| -rw-r--r-- | candle-examples/examples/falcon/model.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/falcon/model.rs b/candle-examples/examples/falcon/model.rs index 1c77cbaf..b638dd51 100644 --- a/candle-examples/examples/falcon/model.rs +++ b/candle-examples/examples/falcon/model.rs @@ -1,6 +1,6 @@ use anyhow::Result; use candle::{DType, Device, Tensor, D}; -use candle_nn::{Embedding, LayerNorm, Linear, VarBuilder}; +use candle_nn::{Embedding, LayerNorm, Linear, Module, VarBuilder}; const MAX_SEQ_LEN: usize = 5000; |