diff options
Diffstat (limited to 'candle-book/src/inference/hub.md')
| -rw-r--r-- | candle-book/src/inference/hub.md | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-book/src/inference/hub.md b/candle-book/src/inference/hub.md index b924b76d..4bd69c14 100644 --- a/candle-book/src/inference/hub.md +++ b/candle-book/src/inference/hub.md @@ -58,7 +58,7 @@ Now that we have our weights, we can use them in our bert architecture: # # let weights = repo.get("model.safetensors").unwrap(); use candle_core::{Device, Tensor, DType}; -use candle_nn::Linear; +use candle_nn::{Linear, Module}; let weights = candle_core::safetensors::load(weights, &Device::Cpu).unwrap(); |